diff --git a/main/flaredetector/flaredetector.py b/main/flaredetector/flaredetector.py index 219ee71..bbfe853 100644 --- a/main/flaredetector/flaredetector.py +++ b/main/flaredetector/flaredetector.py @@ -50,10 +50,12 @@ def calculateFlareFitsForLightcurve(lightcurve, num=100, distance=1, height=0.05 Peak, flareFitDataPoints, fit = calculateFitForFlare(lightcurve, index) peaks.append({"FlarePeak": Peak, "FlarePeakTime": lightcurve.time[index], - "StandardIndex": index}) + "StandardIndex": index, + "Quality": lightcurve.quality[index]}) fits.append({"FlareFit": fit, "FlareFitTime": lightcurve.time[flareFitDataPoints], - "StandardIndex": flareFitDataPoints}) + "StandardIndex": flareFitDataPoints, + "Quality": lightcurve.quality[flareFitDataPoints]}) except Exception as error: pass #print("Flare fit exception: ", error)