fix pstore.models.metadata

- fix for key_0 not in join result
pastas · Feb 7, 2025 · fe6d36c · fe6d36c
1 parent 2cdc095
commit fe6d36c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pastastore/base.py b/pastastore/base.py
@@ -1477,9 +1477,10 @@ def metadata(self):
             names=["oseries", "modelname"],
         )
         modeldf = pd.DataFrame(index=idx)
-        modeldf = modeldf.join(
-            self.conn.oseries, on=modeldf.index.get_level_values(0)
-        ).drop("key_0", axis=1)
+        modeldf = modeldf.join(self.conn.oseries, on=modeldf.index.get_level_values(0))
+        # drop key_0 column if it exists
+        if "key_0" in modeldf.columns:
+            modeldf.drop("key_0", axis=1, inplace=True)
         modeldf["n_stressmodels"] = 0
         for onam, mlnam in modeldf.index:
             mldict = self.conn.get_models(mlnam, return_dict=True)