From a09868010bd0a1ca32f468c38ba031cae2f883a4 Mon Sep 17 00:00:00 2001
From: Alexander Gaponov <67455670+StalkerShurik@users.noreply.github.com>
Date: Wed, 30 Oct 2024 17:06:11 +0300
Subject: [PATCH] fix assert in pit.py (#2811)

(cherry picked from commit 429556b6583deb49a045686a27c0bda8c830f675)
---
 src/torchmetrics/functional/audio/pit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/torchmetrics/functional/audio/pit.py b/src/torchmetrics/functional/audio/pit.py
index 8fa186299f9..a7cc72d48f7 100644
--- a/src/torchmetrics/functional/audio/pit.py
+++ b/src/torchmetrics/functional/audio/pit.py
@@ -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")