Minor fix for GPStime in injection

transientlunatic · Oct 4, 2024 · 377ecbb · 377ecbb
1 parent 62ce6c1
commit 377ecbb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/heron/injection.py b/heron/injection.py
@@ -115,7 +115,7 @@ def injection_parameters_add_units(parameters):
     UNITS = {"luminosity_distance": u.megaparsec, "m1": u.solMass, "m2": u.solMass}
 
     for parameter, value in parameters.items():
-        if not isinstance(value, u.Quantity) and parameter in UNITS:
+        if not isinstance(value, u.Quantity) and (parameter in UNITS.keys()):
             parameters[parameter] = value * UNITS[parameter]
     return parameters