diff --git a/main/flaredetector/flaredetector.py b/main/flaredetector/flaredetector.py index 6064299..e0ebd1e 100644 --- a/main/flaredetector/flaredetector.py +++ b/main/flaredetector/flaredetector.py @@ -30,13 +30,16 @@ def calculateFlareFitsForLightcurve(lightcurve, num=100, distance=1, height=0.05 peaks = [] fits = [] for index in maxIndices: - Peak, flareFitDataPoints, fit = calculateFitForFlare(lightcurve, index) - peaks.append({"FlarePeak": Peak, - "FlarePeakTime": lightcurve.time[index], - "StandardIndex": index}) - fits.append({"FlareFit": fit, - "FlareFitTime": lightcurve.time[flareFitDataPoints], - "StandardIndex": flareFitDataPoints}) + try: + Peak, flareFitDataPoints, fit = calculateFitForFlare(lightcurve, index) + peaks.append({"FlarePeak": Peak, + "FlarePeakTime": lightcurve.time[index], + "StandardIndex": index}) + fits.append({"FlareFit": fit, + "FlareFitTime": lightcurve.time[flareFitDataPoints], + "StandardIndex": flareFitDataPoints}) + except Exception as error: + print("Flare fit exception: ", error) return peaks, fits