fix: optional optuna Journal import

mdtanker · Nov 30, 2023 · e0a159f · e0a159f
1 parent e7ed8a1
commit e0a159f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/invert4geom/optimization.py b/src/invert4geom/optimization.py
@@ -13,7 +13,6 @@
 import harmonica as hm
 import pandas as pd
 from nptyping import NDArray
-from optuna.storages import JournalFileStorage, JournalStorage
 
 from invert4geom import utils
 
@@ -464,7 +463,9 @@ def optimize_eq_source_params(
         file_storage = optuna.storages.JournalFileStorage(fname, lock_obj=lock_obj)
         storage = optuna.storages.JournalStorage(file_storage)
 
-        storage = JournalStorage(JournalFileStorage(fname))
+        storage = optuna.storages.JournalStorage(
+            optuna.storages.JournalFileStorage(fname)
+        )
 
     # if sampler not provided, used BoTorch as default
     if sampler is None: