From 1bc873b90154c6023235d85774ac1b96bc5e1b25 Mon Sep 17 00:00:00 2001 From: marcel Date: Sun, 29 Oct 2023 16:43:13 +0100 Subject: [PATCH] change return type of 'psf_predict' --- pypsf/predict.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pypsf/predict.py b/pypsf/predict.py index 4f99ae3..209b433 100755 --- a/pypsf/predict.py +++ b/pypsf/predict.py @@ -1,4 +1,5 @@ from collections import Counter +from typing import List import numpy as np @@ -8,7 +9,7 @@ def psf_predict(dataset: np.array, n_ahead: int, cycle_length: int, k, w, - supress_warnings=False) -> np.array: + supress_warnings=False) -> List[np.array]: """ Run the PSF algorithm on the provided data to generate the desired number of predictions. @@ -67,4 +68,4 @@ def psf_predict(dataset: np.array, n_ahead: int, cycle_length: int, k, w, psf_warn("No pattern was found in training for any window size.\n" "Using centroid of largest cluster as the prediction!") - return np.array(temp[-n_ahead_cycles:]) + return temp[-n_ahead_cycles:]