owsilhouetteplot: Use any discrete or string var for annotations

biolab · Dec 28, 2021 · 90b01a6 · 90b01a6
1 parent dc6fbc2
commit 90b01a6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Orange/widgets/visualize/owsilhouetteplot.py b/Orange/widgets/visualize/owsilhouetteplot.py
@@ -262,7 +262,8 @@ def _setup_control_models(self, domain: Domain):
             self.cluster_var_idx = groupvars.index(domain.class_var)
         else:
             self.cluster_var_idx = 0
-        annotvars = [var for var in domain.metas if var.is_string]
+        annotvars = [var for var in domain.variables + domain.metas
+                     if var.is_string or var.is_discrete]
         self.annotation_var_model[:] = ["None"] + annotvars
         self.annotation_var_idx = 1 if annotvars else 0
         self.openContext(Orange.data.Domain(groupvars))