Fix duration calculation in Convergence_checker_optuna

RichieHakim · Mar 24, 2024 · f196f8a · f196f8a
1 parent b3ccbf5
commit f196f8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bnpm/optimization.py b/bnpm/optimization.py
@@ -307,7 +307,7 @@ def check(
             else:
                 duration = 0
         elif self.duration_type == 'trials':
-            duration = sum([t.duration.total_seconds() for t in study.trials])
+            duration = sum([t.duration.total_seconds() for t in study.trials if t.duration is not None])
         else:
             raise ValueError(f"duration_type '{self.duration_type}' not recognized")