wums 0.1.9__py3-none-any.whl → 0.1.10__py3-none-any.whl

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/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.dist-info → wums-0.1.10.dist-info}/METADATA

@@ -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.dist-info → wums-0.1.10.dist-info}/RECORD

@@ -1,14 +1,14 @@
 wums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wums/boostHistHelpers.py,sha256=KMZXVPLlx7_v0XgGEQg6mZmuzEOQ6AWO5Wd_mPaAPjg,41636
+wums/boostHistHelpers.py,sha256=AIkJ8QttysAzlH-yoEyS5pV1PwzcVS14fUUGv-zBUts,42163
 wums/fitutils.py,sha256=sPCMJqZGdXvDfc8OxjOB-Bpf45GWHKxmKkDV3SlMUQs,38297
 wums/fitutilsjax.py,sha256=HE1AcIZmI6N_xIHo8OHCPaYkHSnND_B-vI4Gl3vaUmA,2659
 wums/ioutils.py,sha256=EyCOBin7ifStLtdgKl7J1_0VWB6RnXWtIWFge9x73Ow,12465
 wums/logging.py,sha256=L4514Xyq7L1z77Tkh8KE2HX88ZZ06o6SSRyQo96DbC0,4494
 wums/output_tools.py,sha256=89rQPOWpwGuzJK5ZQBDv38rmO9th1D2206QOe9PE-gY,7572
-wums/plot_tools.py,sha256=0olJuXAnuDDXJrbF9GF_ZaWBJ14ljOzF5ww19VGCG_g,54662
+wums/plot_tools.py,sha256=YcMoYiUt-YNntE239vPYL7AMp7HA5tdI-Bz1fHQVm-o,54719
 wums/tfutils.py,sha256=9efkkvxH7VtwJN2yBS6_-P9dLKs3CXdxMFdrEBNsna8,2892
 wums/Templates/index.php,sha256=9EYmfc0ltMqr5oOdA4_BVIHdSbef5aA0ORoRZBEADVw,4348
-wums-0.1.9.dist-info/METADATA,sha256=TgZpEIJTjE_xZ3J_cbrXaD2EClyo3iEKS6MHl5lanLU,1784
-wums-0.1.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-wums-0.1.9.dist-info/top_level.txt,sha256=DCE1TVg7ySraosR3kYZkLIZ2w1Pwk2pVTdkqx6E-yRY,5
-wums-0.1.9.dist-info/RECORD,,
+wums-0.1.10.dist-info/METADATA,sha256=NPyB3B2uOKwH_LGXPA7q91jx83rHA0Vn8F6dXlnC1LE,1785
+wums-0.1.10.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+wums-0.1.10.dist-info/top_level.txt,sha256=DCE1TVg7ySraosR3kYZkLIZ2w1Pwk2pVTdkqx6E-yRY,5
+wums-0.1.10.dist-info/RECORD,,

{wums-0.1.9.dist-info → wums-0.1.10.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any