Update sklearn/utils/multiclass.py

saskra · jeremiedbb · web-flow · commit fdb86242e9c2 · 2025-07-02T13:51:45.000+02:00
Co-authored-by: Jérémie du Boisberranger &lt;jeremie@probabl.ai&gt;
diff --git a/sklearn/utils/multiclass.py b/sklearn/utils/multiclass.py
@@ -414,11 +414,7 @@ def _raise_or_return():
     if issparse(first_row_or_val):
         first_row_or_val = first_row_or_val.data
     classes = cached_unique(y)
-    if (
-        not suppress_warning
-        and y.shape[0] > 20
-        and classes.shape[0] > round(0.5 * y.shape[0])
-    ):
+    if y.shape[0] > 20 and y.shape[0] > classes.shape[0] > round(0.5 * y.shape[0]):
         # Only raise the warning when we have at least 20 samples.
         warnings.warn(
             "The number of unique classes is greater than 50% of the number "
