bugfix: display(): non-standard extension names can be displayed

GeminiDRSoftware · Aug 29, 2023 · 37565b2 · 37565b2
1 parent 159439b
commit 37565b2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/geminidr/core/primitives_visualize.py b/geminidr/core/primitives_visualize.py
@@ -193,8 +193,11 @@ def display(self, adinputs=None, **params):
                 ext.operate(np.squeeze)
 
                 # Get the data we're going to display. TODO Replace extname with attr?
-                data = getattr(ext, {'SCI':'data', 'DQ':'mask',
-                                    'VAR':'variance'}[extname], None)
+                try:
+                    data = getattr(ext, {'SCI':'data', 'DQ':'mask',
+                                         'VAR':'variance'}[extname])
+                except KeyError:
+                    data = getattr(ext, extname, None)
                 dqdata = ext.mask
                 if data is None:
                     log.warning("No data to display in {}[{}]".format(ext.filename,