From 018aa5cbadacc8fbb0f32c5dfc1f8716032436bd Mon Sep 17 00:00:00 2001 From: cardosofede Date: Thu, 7 Dec 2023 17:18:00 -0300 Subject: [PATCH] (feat) add db_path optional --- utils/optuna_database_manager.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/utils/optuna_database_manager.py b/utils/optuna_database_manager.py index abd2318..a124153 100644 --- a/utils/optuna_database_manager.py +++ b/utils/optuna_database_manager.py @@ -1,5 +1,6 @@ import os import json +from typing import Optional import pandas as pd from sqlalchemy import create_engine, text @@ -9,9 +10,10 @@ from utils.data_manipulation import StrategyData class OptunaDBManager: - def __init__(self, db_name): + def __init__(self, db_name, db_root_path: Optional[str]): + db_root_path = db_root_path or "data/backtesting" self.db_name = db_name - self.db_path = f'sqlite:///{os.path.join("data/backtesting", db_name)}' + self.db_path = f'sqlite:///{os.path.join(db_root_path, db_name)}' self.engine = create_engine(self.db_path, connect_args={'check_same_thread': False}) self.session_maker = sessionmaker(bind=self.engine)