Skip to content

Commit 7ee6449

Browse files
authored
fix: should_prune() takes no argument
1 parent 2ad6a19 commit 7ee6449

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pyannote/pipeline/optimizer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -255,8 +255,8 @@ def objective(trial: Trial) -> float:
255255
continue
256256

257257
trial.report(np.mean(losses) if metric is None else abs(metric), i)
258-
if trial.should_prune(i):
259-
raise optuna.structs.TrialPruned()
258+
if trial.should_prune():
259+
raise optuna.TrialPruned()
260260

261261
if show_progress != False:
262262
progress_bar.close()

0 commit comments

Comments
 (0)