Skip to content

Commit

Permalink
fix assert in pit.py (#2811)
Browse files Browse the repository at this point in the history
  • Loading branch information
StalkerShurik authored Oct 30, 2024
1 parent 0a19c47 commit 429556b
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 429556b

Please sign in to comment.