fix logic

PEtab-dev · Jan 8, 2025 · 8b23e88 · 8b23e88
1 parent 2bdda7f
commit 8b23e88
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/petab_select/model_subspace.py b/petab_select/model_subspace.py
@@ -733,15 +733,16 @@ def from_definition(
             The model subspace.
         """
         model_subspace_id = definition.pop(MODEL_SUBSPACE_ID)
-        petab_yaml = definition.pop(MODEL_SUBSPACE_PETAB_YAML)
         if "petab_yaml" in definition:
             petab_yaml = definition.pop("petab_yaml")
             warnings.warn(
                 "Change the `petab_yaml` column to "
                 "`model_subspace_petab_yaml`, in the model space TSV.",
                 DeprecationWarning,
-                stacklevel=2,
+                stacklevel=1,
             )
+        else:
+            petab_yaml = definition.pop(MODEL_SUBSPACE_PETAB_YAML)
         parameters = {
             column_id: decompress_parameter_values(value)
             for column_id, value in definition.items()