Coverage for sparkle/configurator/implementations/__init__.py: 90%

1"""This package provides specific Configurator class implementations for Sparkle.""" 

2from sparkle.configurator.configurator import Configurator 

3from sparkle.configurator.implementations.irace import IRACE, IRACEScenario 

4from sparkle.configurator.implementations.paramils import ParamILS, ParamILSScenario 

5from sparkle.configurator.implementations.smac2 import SMAC2, SMAC2Scenario 

6from sparkle.configurator.implementations.smac3 import SMAC3, SMAC3Scenario 

7 

8 

9def resolve_configurator(configurator_name: str) -> Configurator: 

10 """Returns the Configurator subclass by name.""" 

11 subclass_names = [IRACE.__name__, ParamILS.__name__, SMAC2.__name__, SMAC3.__name__] 

12 if configurator_name in subclass_names: 

13 return eval(configurator_name) 

14 return None