Merge pull request #6533 from noahnovsak/dask-fix-testandscore

Test & Score - fix data error conditions
biolab · Nov 6, 2023 · fbce8ba · fbce8ba
2 parents f655849 + bac8818
commit fbce8ba
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: