ENH support extra extension names

esheldon · Mar 24, 2021 · b3a0398 · b3a0398
1 parent b955ccb
commit b3a0398
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/meds/meds.py b/meds/meds.py
@@ -1055,7 +1055,11 @@ def _get_extension_name(self, type):
         elif type == "noise":
             return "noise_cutouts"
         else:
-            raise ValueError("bad cutout type '%s'" % type)
+            ext = "%s_cutouts" % type
+            if ext not in self._fits:
+                raise ValueError("bad cutout type '%s'" % type)
+            else:
+                return ext
 
     def _check_indices(self, iobj, icutout=None):
         if iobj >= self._cat.size: