Fix Prior.from_par_dict for missing priorParameters columns (PEtab-de…

…v#341) Previously, missing `*PriorParameters` would have resulted in a KeyError.
dweindl · Dec 12, 2024 · 6fffafb · 6fffafb
1 parent 540710d
commit 6fffafb
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/petab/v1/priors.py b/petab/v1/priors.py
@@ -224,10 +224,8 @@ def from_par_dict(
             dist_type = C.PARAMETER_SCALE_UNIFORM
 
         pscale = d.get(C.PARAMETER_SCALE, C.LIN)
-        if (
-            pd.isna(d[f"{type_}PriorParameters"])
-            and dist_type == C.PARAMETER_SCALE_UNIFORM
-        ):
+        params = d.get(f"{type_}PriorParameters", None)
+        if pd.isna(params) and dist_type == C.PARAMETER_SCALE_UNIFORM:
             params = (
                 scale(d[C.LOWER_BOUND], pscale),
                 scale(d[C.UPPER_BOUND], pscale),
@@ -236,7 +234,7 @@ def from_par_dict(
             params = tuple(
                 map(
                     float,
-                    d[f"{type_}PriorParameters"].split(C.PARAMETER_SEPARATOR),
+                    params.split(C.PARAMETER_SEPARATOR),
                 )
             )
         return Prior(