Coverage for agentlib_flexquant/__init__.py: 100%

1"""agentlib plugin""" 

2 

3from agentlib.utils.plugin_import import ModuleImport 

4from agentlib_mpc.optimization_backends import BackendImport, backend_types 

5 

6from .modules import shadow_mpc 

7from .modules import baseline_mpc 

8from .modules import flexibility_indicator 

9from .modules import flexibility_market 

10 

11__version__ = "0.1.0" 

12 

13MODULE_TYPES = { 

14 "shadow_mpc": ModuleImport( 

15 import_path="agentlib_flexquant.modules.shadow_mpc", 

16 class_name=shadow_mpc.FlexibilityShadowMPC.__name__, 

17 ), 

18 "shadow_minlp_mpc": ModuleImport( 

19 import_path="agentlib_flexquant.modules.shadow_mpc", 

20 class_name=shadow_mpc.FlexibilityShadowMINLPMPC.__name__, 

21 ), 

22 "baseline_mpc": ModuleImport( 

23 import_path="agentlib_flexquant.modules.baseline_mpc", 

24 class_name=baseline_mpc.FlexibilityBaselineMPC.__name__, 

25 ), 

26 "baseline_minlp_mpc": ModuleImport( 

27 import_path="agentlib_flexquant.modules.baseline_mpc", 

28 class_name=baseline_mpc.FlexibilityBaselineMINLPMPC.__name__, 

29 ), 

30 "flexibility_indicator": ModuleImport( 

31 import_path="agentlib_flexquant.modules.flexibility_indicator", 

32 class_name=flexibility_indicator.FlexibilityIndicatorModule.__name__, 

33 ), 

34 "flexibility_market": ModuleImport( 

35 import_path="agentlib_flexquant.modules.flexibility_market", 

36 class_name=flexibility_market.FlexibilityMarketModule.__name__, 

37 ), 

38} 

39 

40 

41backend_types.update( 

42 { 

43 "casadi_cia_cons": BackendImport( 

44 import_path="agentlib_flexquant.optimization_backends.constrained_cia", 

45 class_name="ConstrainedCasADiCIABackend", 

46 ) 

47 } 

48)