Merge pull request #555 from PyFstat/transient-predict-fix

get_predict_fstat_parameters_from_dict(): deal with both naming conventions for transient params
PyFstat · Jul 2, 2024 · 11eff6a · 11eff6a
2 parents 9a395e6 + c8f638b
commit 11eff6a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pyfstat/utils/predict.py b/pyfstat/utils/predict.py
@@ -177,9 +177,11 @@ def get_predict_fstat_parameters_from_dict(signal_parameters, transientWindowTyp
         "transient_duration": "transientTau",
     }
     predict_fstat_params = {key: signal_parameters[key] for key in required_keys}
-    for key in transient_keys:
+    for key, val in transient_keys.items():
         if key in signal_parameters:
-            predict_fstat_params[transient_keys[key]] = signal_parameters[key]
+            predict_fstat_params[val] = signal_parameters[key]
+        elif val in signal_parameters:
+            predict_fstat_params[val] = signal_parameters[val]
     if transientWindowType is not None:
         predict_fstat_params["transientWindowType"] = transientWindowType
     return predict_fstat_params