bins when 1D array in test params

sktime · Jul 4, 2024 · 0fcd7c4 · 0fcd7c4
1 parent 017dacc
commit 0fcd7c4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/skpro/regression/binned/_sklearn_bin_regressor.py b/skpro/regression/binned/_sklearn_bin_regressor.py
@@ -262,9 +262,11 @@ def get_test_params(cls, parameter_set="default"):
             `create_test_instance` uses the first (or only) dictionary in `params`
         """
         from sklearn.naive_bayes import GaussianNB
+        from sklearn.semi_supervised import LabelSpreading
         from sklearn.tree import DecisionTreeClassifier
 
         param1 = {"clf": DecisionTreeClassifier(), "bins": 4}
         param2 = {"clf": GaussianNB()}
+        params3 = {"clf": LabelSpreading(), "bins": [20, 80, 160, 250, 300, 380, 420]}
 
-        return [param1, param2]
+        return [param1, param2, params3]