remove cube from viewer creator when no cube data remaining

spacetelescope · Mar 27, 2024 · 5e74ebe · 5e74ebe
1 parent 97e6f20
commit 5e74ebe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lcviz/plugins/viewer_creator/viewer_creator.py b/lcviz/plugins/viewer_creator/viewer_creator.py
@@ -40,7 +40,7 @@ def _rebuild_available_viewers(self, *args):
 
         self.viewer_types = [v for v in self.viewer_types if v['name'].startswith('lcviz')
                              and not v['label'].startswith('flux-vs-phase')
-                             and not v['label'] == 'cube'] + phase_viewers + cube_viewers
+                             and not v['label'] in ('cube', 'image')] + phase_viewers + cube_viewers
         self.send_state('viewer_types')
 
     def vue_create_viewer(self, name):