irrelevant if no TPF data

spacetelescope · Jun 13, 2024 · b7fe196 · b7fe196
1 parent 196a4c6
commit b7fe196
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/lcviz/plugins/photometric_extraction/photometric_extraction.py b/lcviz/plugins/photometric_extraction/photometric_extraction.py
@@ -42,13 +42,21 @@ def __init__(self, *args, **kwargs):
         def is_tpf(data):
             return len(data.shape) == 3
         self.dataset.add_filter(is_tpf)
+        self._set_relevant()
 
     @property
     def user_api(self):
         expose = ['show_live_preview', 'dataset',
                   'add_results', 'extract']
         return PluginUserApi(self, expose=expose)
 
+    @observe('dataset_items')
+    def _set_relevant(self, *args):
+        if len(self.dataset_items) < 1:
+            self.irrelevant_msg = 'Requires at least one TPF cube to be loaded'
+        else:
+            self.irrelevant_msg = ''
+
     @property
     def marks(self):
         marks = {}

diff --git a/lcviz/plugins/stitch/stitch.py b/lcviz/plugins/stitch/stitch.py
@@ -45,14 +45,15 @@ def __init__(self, *args, **kwargs):
         self.dataset.add_filter(data_not_folded, is_not_tpf)
 
         self.results_label_default = 'stitched'
+        self._set_relevant()
 
     @property
     def user_api(self):
         expose = ['dataset', 'stitch', 'remove_input_datasets', 'add_results']
         return PluginUserApi(self, expose=expose)
 
     @observe('dataset_items')
-    def _set_relevent(self, *args):
+    def _set_relevant(self, *args):
         if len(self.dataset_items) < 2:
             self.irrelevant_msg = 'Requires at least two datasets loaded into viewers'
         else: