fix(_find_ml_task): Deal with case where estimator is unfitted and `y…

…` is None
probabl-ai · Jan 16, 2025 · 112167f · 112167f
1 parent a3715db
commit 112167f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/skore/src/skore/sklearn/find_ml_task.py b/skore/src/skore/sklearn/find_ml_task.py
@@ -41,6 +41,9 @@ def _find_ml_task(y, estimator=None) -> MLTask:
                     if estimator.classes_.size > 2:
                         return "multiclass-classification"
             else:  # fallback on the target
+                if y is None:
+                    return "unsupported"
+
                 target_type = type_of_target(y)
                 if target_type == "binary":
                     return "binary-classification"