Merge pull request #27 from cms-cat/lumi_unit

cms-cat · Feb 8, 2024 · 90d21c6 · 90d21c6
2 parents b91cc38 + eda0119
commit 90d21c6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/cmsstyle/cmsstyle.py b/src/cmsstyle/cmsstyle.py
@@ -17,11 +17,11 @@ def SetEnergy(energy):
     cms_energy = str(energy)
 
 
-def SetLumi(lumi, round_lumi=False):
+def SetLumi(lumi, unit="fb", round_lumi=False):
     global cms_lumi
     if lumi != "":
         cms_lumi = f"{lumi:.0f}" if round_lumi else f"{lumi}"
-        cms_lumi += " fb^{#minus1}"
+        cms_lumi += f" {unit}^{{#minus1}}"
     else:
         cms_lumi = lumi