Change error string in functional.py (#694)

NitzanMadar · web-flow · commit b3cbb7548509 · 2022-12-14T08:31:16.000Z
diff --git 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}.")

Original file line number	Diff line number	Diff line change
`@@ -121,7 +121,7 @@ def get_stats(`
`121`	`121`	`)`
`122`	`122`
`123`	`123`	`if torch.is_floating_point(output) and mode == "multiclass":`
`124`		- raise ValueError(f"For ``multiclass`` mode ``target`` should be one of the integer types, got {output.dtype}.")
	`124`	+ raise ValueError(f"For ``multiclass`` mode ``output`` should be one of the integer types, got {output.dtype}.")
`125`	`125`
`126`	`126`	`if mode not in {"binary", "multiclass", "multilabel"}:`
`127`	`127`	raise ValueError(f"``mode`` should be in ['binary', 'multiclass', 'multilabel'], got mode={mode}.")