Merge pull request #1928 from janezd/tree-eq-num-val-bin

[FIX] Tree: Fix crash when two attributes equal number of values (cherry picked from commit c3a6b16)
biolab · Feb 3, 2017 · bda7df4 · bda7df4
1 parent f1d9ccf
commit bda7df4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Orange/widgets/classify/owclassificationtree.py b/Orange/widgets/classify/owclassificationtree.py
@@ -99,13 +99,13 @@ def check_data(self):
         if not self.binary_trees:
             return True
         max_values, max_attr = max(
-            ((len(attr.values), attr)
+            ((len(attr.values), attr.name)
              for attr in self.data.domain.attributes if attr.is_discrete),
             default=(0, None))
         MAX_BINARIZATION = TreeLearner.MAX_BINARIZATION
         if max_values > MAX_BINARIZATION:
             self.Error.cannot_binarize(
-                max_attr.name, max_values, MAX_BINARIZATION)
+                max_attr, max_values, MAX_BINARIZATION)
             return False
         return True