fix data error conditions in owtestandscore for dask

biolab · Aug 16, 2023 · 5aa667e · 5aa667e
1 parent c1aa47f
commit 5aa667e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Orange/widgets/evaluate/owtestandscore.py b/Orange/widgets/evaluate/owtestandscore.py
@@ -414,12 +414,12 @@ def set_train_data(self, data):
                     "Train data input requires a target variable.",
                     not data.domain.class_vars
                 ),
-                ("Target variable has no values.", np.isnan(data.Y).all()),
+                ("Target variable has no values.", np.asarray(np.isnan(data.Y)).all()),
                 (
                     "Target variable has only one value.",
-                    data.domain.has_discrete_class and len(unique(data.Y)) < 2
+                    data.domain.has_discrete_class and len(np.asarray(unique(data.Y))) < 2
                 ),
-                ("Data has no features to learn from.", data.X.shape[1] == 0),
+                ("Data has no features to learn from.", len(data.domain.attributes) == 0),
             ]
 
             for error_msg, cond in data_errors: