diff --git a/datalab/datalab_session/data_operations/rgb_stack.py b/datalab/datalab_session/data_operations/rgb_stack.py index 8cbc570..0f7514a 100644 --- a/datalab/datalab_session/data_operations/rgb_stack.py +++ b/datalab/datalab_session/data_operations/rgb_stack.py @@ -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)