Merge pull request #934 from janezd/single-class

Paint data: Do not output class column if only a single class is present
biolab · Dec 19, 2015 · d4d2c9f · d4d2c9f
2 parents 7b8dd4f + 9369a0a
commit d4d2c9f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Orange/widgets/data/owpaintdata.py b/Orange/widgets/data/owpaintdata.py
@@ -1187,7 +1187,7 @@ def commit(self):
         else:
             X, Y = self.data[:, np.newaxis, 0], self.data[:, 2]
             attrs = (Orange.data.ContinuousVariable(self.attr1),)
-        if len(self.class_model) > 1:
+        if len(np.unique(Y)) >= 2:
             domain = Orange.data.Domain(
                 attrs,
                 Orange.data.DiscreteVariable(