fix: add magnitude type column to empty catalog so that de-/serializa…

…tion works
swiss-seismological-service · Aug 5, 2024 · bf77418 · bf77418
1 parent a010548
commit bf77418
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/seismostats/catalogs/catalog.py b/seismostats/catalogs/catalog.py
@@ -192,7 +192,7 @@ def from_dict(cls,
             df = Catalog(df)
 
         if df.empty:
-            df = Catalog(columns=REQUIRED_COLS_CATALOG)
+            df = Catalog(columns=REQUIRED_COLS_CATALOG + ['magnitude_type'])
 
         return df
 

diff --git a/seismostats/catalogs/tests/test_catalog.py b/seismostats/catalogs/tests/test_catalog.py
@@ -176,7 +176,8 @@ def test_empty_catalog():
 
     catalog = Catalog.from_dict({})
     assert catalog.empty
-    assert catalog.columns.tolist() == REQUIRED_COLS_CATALOG
+    assert catalog.columns.tolist() == REQUIRED_COLS_CATALOG + \
+        ['magnitude_type']
 
     catalog = Catalog.from_dict({'magnitude': []}, include_ids=False)
     assert isinstance(catalog, Catalog)