diff --git a/main/flaredetector/util.py b/main/flaredetector/util.py index f3f56b4..7bfe256 100644 --- a/main/flaredetector/util.py +++ b/main/flaredetector/util.py @@ -152,14 +152,14 @@ def getFoldedFitPeakValley(sineFit): if(len(minima) == 0 and len(maxima) == 1): if(sineFit[0] < sineFit[-1]): - minima = np.array((0)) + minima = np.array((0), ndmin=1) else: - minima = np.array((len(sineFit)-1)) + minima = np.array((len(sineFit)-1), ndmin=1) elif(len(minima) == 1 and len(maxima) == 0): if(sineFit[0] > sineFit[-1]): - minima = np.array((0)) + minima = np.array((0), ndmin=1) else: - minima = np.array((len(sineFit)-1)) + minima = np.array((len(sineFit)-1), ndmin=1) return minima, maxima