freedak/timesfm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #285 from AzulGarza/feat/allow-more-freqs

Browse Source 
feat: allow more frequencies
This commit is contained in:
Rajat Sen
2025-07-06 15:05:16 -07:00
committed by GitHub
parent 5950ef0653 fffb37852b
commit 756217d03b
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/timesfm/timesfm_base.py
+11 -2
View File
@@ -57,9 +57,18 @@ def freq_map(freq: str):
return 1 return 1
elif freq.endswith(("H", "T", "MIN", "D", "B", "U", "S")): elif freq.endswith(("H", "T", "MIN", "D", "B", "U", "S")):
return 0 return 0
elif freq.endswith(("W", "M")): elif (
freq.endswith(("W", "M"))
or freq.startswith("W-")
or (freq.startswith("M") and len(freq) == 2)
):
return 1 return 1
elif freq.endswith(("Y", "Q", "A")): elif (
freq.endswith(("Y", "Q", "A"))
or freq.startswith("Y-")
or freq.startswith("Q-")
or freq.startswith("A-")
):
return 2 return 2
else: else:
raise ValueError(f"Invalid frequency: {freq}") raise ValueError(f"Invalid frequency: {freq}")