Update src/timesfm.py

Co-authored-by: Kashif Rasul <kashif.rasul@gmail.com>
This commit is contained in:
Sarthak Mishra
2024-05-15 19:40:22 +05:30
committed by GitHub
parent 09f585d993
commit 38865205f9
+1 -1
View File
@@ -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