From 33c46d12fad55426b2439861f0df0fa5d9c0b9fe Mon Sep 17 00:00:00 2001 From: Maria Pena-Guerrero Date: Fri, 20 Sep 2024 13:38:48 -0400 Subject: [PATCH] fixing silly mistake --- jwst/saturation/saturation.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/jwst/saturation/saturation.py b/jwst/saturation/saturation.py index 2e8361b29a..dbefe13fce 100644 --- a/jwst/saturation/saturation.py +++ b/jwst/saturation/saturation.py @@ -55,6 +55,8 @@ def flag_saturation(output_model, ref_model, n_pix_grow_sat, use_readpatt): gdq = output_model.groupdq pdq = output_model.pixeldq data = output_model.data + nframes = output_model.meta.exposure.nframes + ngroups = output_model.meta.exposure.ngroups zframe = output_model.zeroframe if output_model.meta.exposure.zero_frame else None @@ -93,7 +95,7 @@ def flag_saturation(output_model, ref_model, n_pix_grow_sat, use_readpatt): return output_model -def irs2_flag_saturation(input_model, ref_model, n_pix_grow_sat, use_readpatt): +def irs2_flag_saturation(output_model, ref_model, n_pix_grow_sat, use_readpatt): """ Short Summary ------------- @@ -104,7 +106,7 @@ def irs2_flag_saturation(input_model, ref_model, n_pix_grow_sat, use_readpatt): Parameters ---------- - input_model : `~jwst.datamodels.RampModel` + output_model : `~jwst.datamodels.RampModel` The input science data to be corrected ref_model : `~jwst.datamodels.SaturationModel`