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 baseline_mpc, flexibility_indicator, flexibility_market, shadow_mpc 

7 

8__version__ = "0.1.0" 

9 

10MODULE_TYPES = { 

11 "shadow_mpc": ModuleImport( 

12 import_path="agentlib_flexquant.modules.shadow_mpc", 

13 class_name=shadow_mpc.FlexibilityShadowMPC.__name__, 

14 ), 

15 "shadow_minlp_mpc": ModuleImport( 

16 import_path="agentlib_flexquant.modules.shadow_mpc", 

17 class_name=shadow_mpc.FlexibilityShadowMINLPMPC.__name__, 

18 ), 

19 "baseline_mpc": ModuleImport( 

20 import_path="agentlib_flexquant.modules.baseline_mpc", 

21 class_name=baseline_mpc.FlexibilityBaselineMPC.__name__, 

22 ), 

23 "baseline_minlp_mpc": ModuleImport( 

24 import_path="agentlib_flexquant.modules.baseline_mpc", 

25 class_name=baseline_mpc.FlexibilityBaselineMINLPMPC.__name__, 

26 ), 

27 "flexibility_indicator": ModuleImport( 

28 import_path="agentlib_flexquant.modules.flexibility_indicator", 

29 class_name=flexibility_indicator.FlexibilityIndicatorModule.__name__, 

30 ), 

31 "flexibility_market": ModuleImport( 

32 import_path="agentlib_flexquant.modules.flexibility_market", 

33 class_name=flexibility_market.FlexibilityMarketModule.__name__, 

34 ), 

35} 

36 

37 

38backend_types.update( 

39 { 

40 "casadi_cia_cons": BackendImport( 

41 import_path="agentlib_flexquant.optimization_backends.constrained_cia", 

42 class_name="ConstrainedCasADiCIABackend", 

43 ) 

44 } 

45)