Correctly handle regular single-hdu fits files

askap-vast · Dec 7, 2023 · bffb97b · bffb97b
1 parent d37f4dc
commit bffb97b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/vast_pipeline/image/utils.py b/vast_pipeline/image/utils.py
@@ -271,7 +271,9 @@ def open_fits(fits_path: Union[str, Path], memmap: Optional[bool] = True):
     hdul = fits.open(fits_path, memmap=memmap)
 
     # This is a messy way to check, but I can't think of a better one
-    if type(hdul[1]) == fits.hdu.compressed.CompImageHDU:
+    if len(hdul) == 1:
+        return hdul
+    elif type(hdul[1]) == fits.hdu.compressed.CompImageHDU:
         return fits.HDUList(hdul[1:])
     else:
         return hdul