From 6eaac93f6f40def8f28c18d1f2f316091c3fe3c0 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Sun, 21 Feb 2021 11:49:56 +0100 Subject: [PATCH] norm_freq_str for fourier_time_features fixes issue #41 --- pts/feature/fourier_date_feature.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pts/feature/fourier_date_feature.py b/pts/feature/fourier_date_feature.py index 18deaa7..1d7556b 100644 --- a/pts/feature/fourier_date_feature.py +++ b/pts/feature/fourier_date_feature.py @@ -18,7 +18,7 @@ import pandas as pd from pandas.tseries.frequencies import to_offset from gluonts.core.component import validated -from gluonts.time_feature import TimeFeature +from gluonts.time_feature import TimeFeature, norm_freq_str class FourierDateFeatures(TimeFeature): @@ -50,7 +50,7 @@ class FourierDateFeatures(TimeFeature): def fourier_time_features_from_frequency(freq_str: str) -> List[TimeFeature]: offset = to_offset(freq_str) - multiple, granularity = offset.n, offset.name + granularity = norm_freq_str(offset.name) features = { "M": ["weekofyear"],