Skip to content

Commit

Permalink
Merge branch 'master' into nisqa
Browse files Browse the repository at this point in the history
  • Loading branch information
SkafteNicki authored Oct 30, 2024
2 parents 23e0d6c + 429556b commit 2b34960
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/torchmetrics/functional/audio/pit.py
Original file line number Diff line number Diff line change
Expand Up @@ -161,7 +161,7 @@ def permutation_invariant_training(
if eval_func not in ["max", "min"]:
raise ValueError(f'eval_func can only be "max" or "min" but got {eval_func}')
if mode not in ["speaker-wise", "permutation-wise"]:
raise ValueError(f'mode can only be "speaker-wise" or "permutation-wise" but got {eval_func}')
raise ValueError(f'mode can only be "speaker-wise" or "permutation-wise" but got {mode}')
if target.ndim < 2:
raise ValueError(f"Inputs must be of shape [batch, spk, ...], got {target.shape} and {preds.shape} instead")

Expand Down

0 comments on commit 2b34960

Please sign in to comment.