Merge pull request #50 from esheldon/any-ext

ENH support extra extension names
esheldon · Mar 24, 2021 · 066cb3e · 066cb3e
2 parents b955ccb + b3a0398
commit 066cb3e
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: