From 88690715bc7b386d624d7f208d1ec7f0e2b6e18b Mon Sep 17 00:00:00 2001 From: BBing <67486385@qq.com> Date: Wed, 4 Oct 2023 12:44:30 +0800 Subject: [PATCH] Update timefeatures.py --- utils/timefeatures.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/timefeatures.py b/utils/timefeatures.py index 2708026..c6ba0b6 100644 --- a/utils/timefeatures.py +++ b/utils/timefeatures.py @@ -35,7 +35,7 @@ class HourOfDay(TimeFeature): class DayOfWeek(TimeFeature): """Hour of day encoded as value between [-0.5, 0.5]""" - def __call__(self, index: pd.DatetimeIndex) -> np.ndarray: + def __call__(self, index: pd.DatetimeIndex) -> mnp.ndarray: return index.dayofweek / 6.0 - 0.5 class DayOfMonth(TimeFeature):