From 09f585d993a5c56e0ce0ba710fb73fcfba772acb Mon Sep 17 00:00:00 2001 From: Sarthak Mishra Date: Tue, 14 May 2024 17:53:40 +0530 Subject: [PATCH] allow month start frequency --- src/timesfm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/timesfm.py b/src/timesfm.py index 8850239..0de4062 100644 --- a/src/timesfm.py +++ b/src/timesfm.py @@ -70,7 +70,7 @@ def freq_map(freq: str): or freq.endswith("U") ): return 0 - elif freq.endswith("W") or freq.endswith("M"): + elif freq.endswith("W") or freq.endswith("M") or freq.endswith("MS"): return 1 elif freq.endswith("Y") or freq.endswith("Q"): return 2