restrict function to Sum

* dropdown will still show until future PR to upstream
spacetelescope · Oct 17, 2024 · a5f024a · a5f024a
1 parent fae2324
commit a5f024a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lcviz/plugins/photometric_extraction/photometric_extraction.py b/lcviz/plugins/photometric_extraction/photometric_extraction.py
@@ -40,11 +40,16 @@ def __init__(self, *args, **kwargs):
         def is_tpf(data):
             return len(data.shape) == 3
         self.dataset.filters = [is_tpf]
+
+        # only allow for Sum
+        self.function._manual_options = ['Sum']
+        self.function.items = [{"label": "Sum"}]
+
         self._set_relevant()  # move upstream?
 
     @property
     def user_api(self):
-        expose = ['dataset', 'function', 'aperture',
+        expose = ['dataset', 'aperture',
                   'background',
                   'add_results', 'extract',
                   'aperture_method']