From 0fb2089aad0bef428527c9f3b81685f807e3b004 Mon Sep 17 00:00:00 2001
From: Nitzan Madar <67631058+NitzanMadar@users.noreply.github.com>
Date: Fri, 9 Dec 2022 10:36:44 +0200
Subject: [PATCH] Change error string in functional.py

---
 segmentation_models_pytorch/metrics/functional.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/segmentation_models_pytorch/metrics/functional.py b/segmentation_models_pytorch/metrics/functional.py
index 435d6947..38f60c1e 100644
--- a/segmentation_models_pytorch/metrics/functional.py
+++ b/segmentation_models_pytorch/metrics/functional.py
@@ -121,7 +121,7 @@ def get_stats(
         )
 
     if torch.is_floating_point(output) and mode == "multiclass":
-        raise ValueError(f"For ``multiclass`` mode ``target`` should be one of the integer types, got {output.dtype}.")
+        raise ValueError(f"For ``multiclass`` mode ``output`` should be one of the integer types, got {output.dtype}.")
 
     if mode not in {"binary", "multiclass", "multilabel"}:
         raise ValueError(f"``mode`` should be in ['binary', 'multiclass', 'multilabel'], got mode={mode}.")