update to remove "optuna" warnings

rlberry-py · Oct 24, 2024 · 0468d8e · 0468d8e
1 parent 289e34f
commit 0468d8e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/rlberry/manager/experiment_manager.py b/rlberry/manager/experiment_manager.py
@@ -647,7 +647,9 @@ def eval_agents(
     def clear_output_dir(self):
         """Delete output_dir and all its data."""
         if self.optuna_study:
-            optuna.delete_study(self.optuna_study.study_name, self.optuna_storage_url)
+            optuna.delete_study(
+                study_name=self.optuna_study.study_name, storage=self.optuna_storage_url
+            )
         try:
             shutil.rmtree(self.output_dir_)
         except FileNotFoundError: