Update label.py

Correctly format com and lumi
scikit-hep · Jul 13, 2023 · 060cfa1 · 060cfa1
1 parent 9894bcb
commit 060cfa1
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/src/mplhep/label.py b/src/mplhep/label.py
@@ -347,15 +347,18 @@ def exp_label(
             fig=ax.figure,
         )
         if lumi is not None:
-            _lumi = (
-                r"$\sqrt{s} = \mathrm{"
-                + lumi_format.format(lumi)
-                + r"\ TeV}, "
-                + str(lumi)
-                + r"\ \mathrm{fb}^{-1}$"
+            _lumi = r"{com}, {lumi}".format(
+                com=r"$\sqrt{s} = \mathrm{" + str(com) + r"\ TeV}$"
+                if com is not None
+                else r"$\sqrt{s} = \mathrm{13\ TeV}$",
+                lumi=lumi_format.format(lumi) + r" $\mathrm{fb^{-1}}$",
             )
         else:
-            _lumi = ""
+            _lumi = r"{com}".format(
+                com=r"$\sqrt{s} = \mathrm{" + str(com) + r"\ TeV}$"
+                if com is not None
+                else r"$\sqrt{s} = \mathrm{13\ TeV}$",
+            )
         explumi = ExpSuffix(
             *exptext.get_position(),
             text=rlabel if rlabel is not None else _lumi,