Allow sm read source_array

RadioAstronomySoftwareGroup · Feb 9, 2024 · 6ddfe01 · 6ddfe01
1 parent f3f91ce
commit 6ddfe01
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/pyradiosky/skymodel.py b/src/pyradiosky/skymodel.py
@@ -4351,7 +4351,13 @@ def read_fhd_catalog(
 
 
         """
-        catalog = scipy.io.readsav(filename_sav)["catalog"]
+        catalog = scipy.io.readsav(filename_sav)
+        if "catalog" in catalog.keys():
+            catalog = catalog["catalog"]
+        elif "source_array" in catalog.keys():
+            catalog = catalog["source_array"]
+        else:
+            raise KeyError(f"File {filename_sav} does not contain a known catalog name")
         ids = catalog["id"].astype(str)
         ra = catalog["ra"]
         dec = catalog["dec"]