diff --git a/main/flaredetector/flaredetector.py b/main/flaredetector/flaredetector.py index e0ebd1e..1ec7bba 100644 --- a/main/flaredetector/flaredetector.py +++ b/main/flaredetector/flaredetector.py @@ -62,6 +62,8 @@ def getFlareRange(flux, peakIndex): if(deltaCount > 3): minInd = i break + else: + deltaCount = 0 prevFlux = flux[i] if(minInd == -np.inf): @@ -83,6 +85,8 @@ def getFlareRange(flux, peakIndex): if(deltaCount > 3): maxInd = i break + else: + deltaCount = 0 prevFlux = flux[i] if(maxInd == np.inf):