fix optuna storage default param

tryolabs · Mar 5, 2024 · 5f273be · 5f273be
1 parent c5210f1
commit 5f273be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pipeline_lib/implementation/tabular/xgboost/fit_model.py b/pipeline_lib/implementation/tabular/xgboost/fit_model.py
@@ -138,7 +138,7 @@ def optuna_logging_callback(study, trial):
         self.logger.info(f"Optimizing XGBoost hyperparameters with {optuna_trials} trials.")
 
         study_name = optuna_params.get("study_name", "xgboost_optimization")
-        storage = optuna_params.get("storage", "I")
+        storage = optuna_params.get("storage", "sqlite:///db.sqlite3")
 
         study = optuna.create_study(
             direction="minimize",