fix: should_prune() takes no argument

pyannote · Jan 12, 2025 · 7ee6449 · 7ee6449
1 parent 2ad6a19
commit 7ee6449
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyannote/pipeline/optimizer.py b/pyannote/pipeline/optimizer.py
@@ -255,8 +255,8 @@ def objective(trial: Trial) -> float:
                     continue
 
                 trial.report(np.mean(losses) if metric is None else abs(metric), i)
-                if trial.should_prune(i):
-                    raise optuna.structs.TrialPruned()
+                if trial.should_prune():
+                    raise optuna.TrialPruned()
 
             if show_progress != False:
                 progress_bar.close()