add z to plots if present

desy-multimessenger · May 10, 2023 · ac26c9e · ac26c9e
1 parent 0e76c55
commit ac26c9e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/nuztf/plot.py b/nuztf/plot.py
@@ -243,9 +243,10 @@ def absmag_to_mag(absmag):
             )
 
         if (redshift := alert[0].get("redshifts", {}).get("ampel_z")) is not None:
-            info.append(
-                f"z: {redshift:.3f} +/- {alert[0]['redshifts']['group_z_precision']}"
-            )
+            if alert[0]["redshifts"]["group_z_nbr"] in [1, 2]:
+                info.append(f"spec z: {redshift:.3f}")
+            else:
+                info.append(f"photo z: {redshift:.3f}")
 
         fig.text(0.77, 0.55, "\n".join(info), va="top", fontsize="medium", alpha=0.5)