From 38865205f953cdc31a5dacd6aac824be847d4e75 Mon Sep 17 00:00:00 2001 From: Sarthak Mishra Date: Wed, 15 May 2024 19:40:22 +0530 Subject: [PATCH] Update src/timesfm.py Co-authored-by: Kashif Rasul --- src/timesfm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/timesfm.py b/src/timesfm.py index 0de4062..33b82e2 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") or freq.endswith("MS"): + elif freq.endswith(("W", "M", "MS")): return 1 elif freq.endswith("Y") or freq.endswith("Q"): return 2