[FIX] objective wrapper for windows

BioSystemsUM · Nov 9, 2024 · 8375eb9 · 8375eb9
1 parent 5ae2406
commit 8375eb9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/deepmol/pipeline_optimization/objective_wrapper.py b/src/deepmol/pipeline_optimization/objective_wrapper.py
@@ -42,8 +42,8 @@ def __init__(self, objective_steps, study, direction, save_top_n, trial_timeout=
     def run_with_timeout_windows(self):
 
         @timeout(self.trial_timeout)
-        def run_with_timeout():
-            return self._run()
+        def run_with_timeout(trial):
+            return self._run(trial)
 
         return run_with_timeout