add to filters for rgb operation, used for sorting input

LCOGT · Feb 11, 2025 · 220d4c0 · 220d4c0
1 parent ffbb461
commit 220d4c0
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/datalab/datalab_session/data_operations/rgb_stack.py b/datalab/datalab_session/data_operations/rgb_stack.py
@@ -45,7 +45,7 @@ def wizard_description():
                     'maximum': 1,
                     'include_custom_scale': True,
                     'combine_custom_scale': 'rgb',
-                    'filter': ['rp', 'r']
+                    'filter': ['rp', 'r', 'ip', 'h-alpha']
                 },
                 'green_input': {
                     'name': 'Green Filter',
@@ -55,7 +55,7 @@ def wizard_description():
                     'maximum': 1,
                     'include_custom_scale': True,
                     'combine_custom_scale': 'rgb',
-                    'filter': ['V', 'gp']
+                    'filter': ['v', 'gp', 'oiii']
                 },
                 'blue_input': {
                     'name': 'Blue Filter',
@@ -65,7 +65,7 @@ def wizard_description():
                     'maximum': 1,
                     'include_custom_scale': True,
                     'combine_custom_scale': 'rgb',
-                    'filter': ['B']
+                    'filter': ['b', 'sii']
                 }
             },
         }
@@ -107,10 +107,9 @@ def _align_images(self, fits_files: list[str]) -> list[str]:
                 aligned_images.append(aligned_img)
 
         if len(aligned_images) != self.REQUIRED_INPUTS:
-            raise ClientAlertException('Failed to align all images')
+            log.info('could not align all images')
+            return fits_files
 
-        self.set_operation_progress(self.PROGRESS_STEPS['ALIGNMENT'])
-
         return aligned_images
 
     # Currently storing the output fits SCI HDU as a 3D ndarray consisting of each input's SCI data
@@ -132,6 +131,7 @@ def operate(self):
         fits_files = [handler.fits_file for handler in input_handlers]
 
         aligned_images = self._align_images(fits_files)
+        self.set_operation_progress(self.PROGRESS_STEPS['ALIGNMENT'])
 
         with create_jpgs(self.cache_key, aligned_images, color=True, zmin=zmin_list, zmax=zmax_list) as (large_jpg_path, small_jpg_path):
             stacked_ndarray = self._create_3d_array(input_handlers)