fix: pyannote#52 - should_prune() takes no argument

arxaqapi · Aug 7, 2024 · 5540157 · 5540157
1 parent 588a203
commit 5540157
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyannote/pipeline/optimizer.py b/pyannote/pipeline/optimizer.py
@@ -253,7 +253,7 @@ def objective(trial: Trial) -> float:
                     continue
 
                 trial.report(np.mean(losses) if metric is None else abs(metric), i)
-                if trial.should_prune(i):
+                if trial.should_prune():
                     raise optuna.structs.TrialPruned()
 
             if show_progress != False: