From f19dba985c20f4f3e1d5de5b3356074826581cad Mon Sep 17 00:00:00 2001 From: SGCMarkus Date: Wed, 17 Apr 2024 09:33:21 +0200 Subject: [PATCH] flaredetector: catch exception if we cant fit the flare --- main/flaredetector/flaredetector.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) 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