Modify checks for categorical axis

scikit-hep · Jun 29, 2021 · 054c1c6 · 054c1c6
1 parent 7d4a8bd
commit 054c1c6
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/hist/stacks.py b/src/hist/stacks.py
@@ -31,9 +31,12 @@ def plot(
         """
         Plot method for Stack object.
         """
-        _has_categorical = (
-            np.sum([ax.traits.discrete for ax in self._stack[0].axes]) == 1
-        )
+        _has_categorical = 0
+        if (
+            np.sum(self._stack[0].axes.traits.ordered) == 1
+            and np.sum(self._stack[0].axes.traits.discrete) == 1
+        ):
+            _has_categorical = 1
         _project = _has_categorical or overlay is not None
         if self._stack[0].ndim == 1 or (self._stack[0].ndim == 2 and _project):
             return [