Coverage for rivapy/marketdata_tools/__init__.py: 91%

1from rivapy.marketdata_tools.factory import _factory 

2from rivapy.marketdata_tools.pfc_shifter import PFCShifter 

3from rivapy.marketdata_tools.pfc_shaper import PFCShaper, CategoricalRegression 

4 

5 

6def _add_to_factory(cls): 

7 factory_entries = _factory() 

8 factory_entries[cls.__name__] = cls 

9 

10 

11_add_to_factory(PFCShifter) 

12_add_to_factory(PFCShaper) 

13_add_to_factory(CategoricalRegression) 

14 

15 

16if __name__ == "__main__": 

17 pass