From 7ee6449e4b001c0df126e89a5a671ade2fe89a45 Mon Sep 17 00:00:00 2001
From: Tarek <45296979+arxaqapi@users.noreply.github.com>
Date: Sun, 12 Jan 2025 21:51:11 +0100
Subject: [PATCH] fix: should_prune() takes no argument

---
 pyannote/pipeline/optimizer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyannote/pipeline/optimizer.py b/pyannote/pipeline/optimizer.py
index bb17323..eb25eee 100644
--- 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()