def __init__(self, name: str, db_path: str,
parameters: Optional[List[TParameterization]] = None,
objective_name: Optional[str] = None):
if not os.path.isfile(db_path):
init_engine_and_session_factory(url=f'sqlite:///{db_path}')
create_all_tables(get_engine())
self.name = name
self.ax = AxClient(enforce_sequential_optimization=False,
verbose_logging=False,
db_settings=DBSettings(url=f'sqlite:///{db_path}'))
if self.ax._experiment is None:
try:
self.ax.create_experiment(name=name, parameters=parameters,
objective_name=objective_name)
except ValueError:
self.ax.load_experiment_from_database(name)
def testCreationOfTestDB(self):
init_test_engine_and_session_factory(tier_or_path=":memory:",
force_init=True)
engine = get_engine()
self.assertIsNotNone(engine)
