wums 0.1.9__tar.gz → 0.1.10__tar.gz

{wums-0.1.9 → wums-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wums
-Version: 0.1.9
+Version: 0.1.10
 Summary: .
 Author-email: David Walter <david.walter@cern.ch>, Josh Bendavid <josh.bendavid@cern.ch>, Kenneth Long <kenneth.long@cern.ch>, Jan Eysermans <jan.eysermans@cern.ch>
 License: MIT

{wums-0.1.9 → wums-0.1.10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "wums"
-version = "0.1.9"
+version = "0.1.10"
 description = "."
 readme = { file = "README.md", content-type = "text/markdown" }
 license = { text = "MIT" }

{wums-0.1.9 → wums-0.1.10}/wums/boostHistHelpers.py

@@ -528,9 +528,27 @@ def disableAxisFlow(ax, under=False, over=False):
         circular=ax.traits.circular,
     )
 
+def enableAxisFlow(ax, overflow=True, underflow=True):
+    if isinstance(ax, hist.axis.Integer):
+        args = [ax.edges[0], ax.edges[-1]]
+    elif isinstance(ax, hist.axis.Regular):
+        args = [ax.size, ax.edges[0], ax.edges[-1]]
+    else:
+        args = [ax.edges]
+
+    return type(ax)(
+        *args,
+        name=ax.name,
+        overflow=overflow,
+        underflow=underflow,
+        circular=ax.traits.circular,
+    )
 
-def disableFlow(h, axis_name, under=False, over=False):
-    # axes_name can be either string or a list of strings with the axis name(s) to disable the flow
+def disableFlow(h, axis_name=None, under=False, over=False):
+    # axes_name can be 'None' for all axes, a string or a list of strings with the axis name(s) to disable the flow
+    if axis_name is None:
+        axis_name = [n for n in h.axes.name]
+    
     if not isinstance(axis_name, str):
         for var in axis_name:
             if var in h.axes.name:

{wums-0.1.9 → wums-0.1.10}/wums/plot_tools.py

@@ -644,7 +644,7 @@ def add_decor(
         make_label(
             ax=ax,
             lumi=lumi,
-            lumi_format="{0:.3g}",
+            lumi_format="{}" if isinstance(lumi, (list, tuple, np.ndarray)) else "{0:.3g}",
             fontsize=text_size,
             label=label,
             data=data,

{wums-0.1.9 → wums-0.1.10}/wums.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wums
-Version: 0.1.9
+Version: 0.1.10
 Summary: .
 Author-email: David Walter <david.walter@cern.ch>, Josh Bendavid <josh.bendavid@cern.ch>, Kenneth Long <kenneth.long@cern.ch>, Jan Eysermans <jan.eysermans@cern.ch>
 License: MIT