Updating the CHARGELOSS read noise computation to only update pixels …

…and integrations affected by CHARGELOSS.
spacetelescope · Mar 14, 2024 · 562322a · 562322a
1 parent 4825e7a
commit 562322a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/jwst/ramp_fitting/ramp_fit_step.py b/jwst/ramp_fitting/ramp_fit_step.py
@@ -479,11 +479,16 @@ def process(self, input):
                 # readnoise variances into copies of the original ramp fitting
                 # tuples.
                 image_info_new, integ_info_new = None, None
+                ch_int, ch_grp, ch_row, ch_col = wh_chargeloss
                 if image_info is not None and image_var_RN is not None:
-                    image_info_new = (image_info[0], image_info[1], image_info[2], image_var_RN, image_info[4])
+                    rnoise = image_info[3]
+                    rnoise[ch_row, ch_col] = image_var_RN[ch_row, ch_col]
+                    image_info_new = (image_info[0], image_info[1], image_info[2], rnoise, image_info[4])
 
                 if integ_info is not None and integ_var_RN is not None:
-                    integ_info_new = (integ_info[0], integ_info[1], integ_info[2], integ_var_RN, integ_info[4])
+                    rnoise = integ_info[3]
+                    rnoise[ch_int, ch_row, ch_col] = integ_var_RN[ch_int, ch_row, ch_col]
+                    integ_info_new = (integ_info[0], integ_info[1], integ_info[2], rnoise, integ_info[4])
 
                 image_info = image_info_new
                 integ_info = integ_info_new