aljaz/hummingbot-dashboard
1
0
Fork 0
mirror of https://github.com/aljazceru/hummingbot-dashboard.git synced 2026-01-16 20:04:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #111 from hummingbot/fix/optuna-db-manager-fixes

Browse Source
This commit is contained in:
dardonacci
2024-01-06 15:39:18 -03:00
committed by GitHub
parent 80850efaff a0ededc98c
commit 2d8086c88e
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pages/backtest_analyze/analyze.py
View File

@@ -19,11 +19,13 @@ from utils.st_utils import initialize_st_page
initialize_st_page(title="Analyze", icon="🔬", initial_sidebar_state="collapsed")
BASE_DATA_DIR = "data/backtesting"
@st.cache_resource
def get_databases():
sqlite_files = [db_name for db_name in os.listdir("data/backtesting") if db_name.endswith(".db")]
databases_list = [OptunaDBManager(db) for db in sqlite_files]
sqlite_files = [db_name for db_name in os.listdir(BASE_DATA_DIR) if db_name.endswith(".db")]
databases_list = [OptunaDBManager(db, db_root_path=BASE_DATA_DIR) for db in sqlite_files]
databases_dict = {database.db_name: database for database in databases_list}
return [x.db_name for x in databases_dict.values() if x.status == 'OK']
@@ -45,7 +47,7 @@ else:
# Select database from selectbox
selected_db = st.selectbox("Select your database:", dbs)
# Instantiate database manager
opt_db = OptunaDBManager(selected_db)
opt_db = OptunaDBManager(selected_db, db_root_path=BASE_DATA_DIR)
# Load studies
studies = opt_db.load_studies()
# Choose study