Added additional handling for integer bins

HEPData · Nov 7, 2023 · f7dcb33 · f7dcb33
1 parent 11ae275
commit f7dcb33
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/hepdata_lib/hist_utils.py b/hepdata_lib/hist_utils.py
@@ -77,12 +77,17 @@ def _get_histaxis_array(axis, flow: bool) -> numpy.ndarray:
             entries.append("__UNDETERMINED__")
         elif isinstance(axis, hist.axis.IntCategory):
             entries.append(entries[-1] + 1)
+        elif isinstance(axis, hist.axis.Integer):
+            entries.append(numpy.inf)
         else:
             entries.append((axis.edges[-1], numpy.inf))
 
     ## Adding underflow bin
     if flow and axis.traits.underflow:
-        entries = [(-numpy.inf, axis.edges[0])] + entries
+        if isinstance(axis,hist.axis.Integer):
+            entries = [-numpy.inf] + entries
+        else:
+            entries = [(-numpy.inf, axis.edges[0])] + entries
 
     ## Converting to numpy array
     if axis.traits.continuous: