From 7205fdba4b795a13ca69954625a689b1c265c70b Mon Sep 17 00:00:00 2001
From: Mike Regan <mregan@stsci.edu>
Date: Fri, 26 Jan 2024 15:31:53 -0500
Subject: [PATCH] Update jump.py

addressed comments in PR
---
 src/stcal/jump/jump.py | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/src/stcal/jump/jump.py b/src/stcal/jump/jump.py
index 66c72fa6..73e5bbe1 100644
--- a/src/stcal/jump/jump.py
+++ b/src/stcal/jump/jump.py
@@ -473,7 +473,7 @@ def detect_jumps(
     data /= gain_2d
     err /= gain_2d
     readnoise_2d /= gain_2d
-     # Return the updated data quality arrays
+    # Return the updated data quality arrays
     return gdq, pdq, total_primary_crs, number_extended_events, stddev
 
 
@@ -684,6 +684,22 @@ def extend_ellipses(
     return out_gdq_cube, num_ellipses
 
 def find_last_grp(grp, ngrps, num_grps_masked_after):
+    """
+    Parameters
+    _________
+    grp : int
+    The location of the shower
+    ngrps :
+    The number of groups in the integration
+    num_grps_masked_after :
+    The requested number of groups to be flagged after the shower
+
+    Returns
+    _______
+    last_grp : int
+      The index of the last group to flag for the shower
+
+    """
     num_grps_masked_after += 1
     last_grp = min(grp + num_grps_masked_after, ngrps)
     return last_grp