diff --git a/main/flaredetector/flaredetector.py b/main/flaredetector/flaredetector.py index 8086d03..58aebe4 100644 --- a/main/flaredetector/flaredetector.py +++ b/main/flaredetector/flaredetector.py @@ -42,6 +42,10 @@ def plotFlarePeaks(ax, peaks): for peak in peaks: ax.plot(peak["FlarePeakTime"], peak["FlarePeak"], "x", color="red") +def plotFlareFits(ax, fits): + for fit in fits: + ax.plot(fit["FlareFitTime"], fit["FlareFit"], "g--") + def getFlareRange(flux, peakIndex): minInd = -np.inf maxInd = np.inf