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)