Skip to content

Add PatchTST-FM support for levels #331

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
spolisar wants to merge 21 commits into
base: main
Choose a base branch
from
Open

Add PatchTST-FM support for levels #331

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
21 commits
Select commit Hold shift + click to select a range
f33e730
feat: init work distributed
AzulGarza Feb 19, 2026
6876edf
feat: not parallel inside dask
AzulGarza Mar 18, 2026
4c5d2ba
feat: update dependencies for distributed support
spolisar Mar 19, 2026
e41b431
ci: add distributed tests to ci
spolisar Mar 19, 2026
ce09abf
fix: repartition single partition dask dataframes
spolisar Mar 19, 2026
748a96a
test: add test for distributed dask edge case
spolisar Mar 19, 2026
b9a80e5
docs: add example notebooks for distributed
spolisar Mar 19, 2026
f64e72e
test: switch to smaller chronos model
spolisar Mar 20, 2026
f1da75a
Merge branch 'main' into feat/anydataframe
spolisar Mar 20, 2026
6a7fd4b
docs: add examples to mkdocs.yml
spolisar Mar 20, 2026
77e3dd1
style: ruff-formatted files
spolisar Mar 20, 2026
1af34a8
style: toml-sorted uv.lock
spolisar Mar 20, 2026
93f9d7b
ci: limit distributed ci tests to 1 worker
spolisar Mar 20, 2026
0494ce8
docs: switch distributed df examples to event_pageviews dataset
spolisar Mar 23, 2026
fd42590
fix: round in level->quantile conversion
spolisar Mar 24, 2026
5d0431c
test: match test to fixed level->quantile conversion
spolisar Mar 24, 2026
b4f5734
test: add tests for levels with distributed dfs
spolisar Mar 24, 2026
fa9d80a
feat: add support for levels with patchTST-FM
spolisar Mar 25, 2026
28bc45f
chore: remove comment, handled in previous commit
spolisar Mar 26, 2026
8b26a2b
Merge branch 'main' into feat/patchtst-levels
AzulGarza Apr 20, 2026
cb1acef
fix: rm duplicated keys
AzulGarza Apr 20, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion tests/models/test_models.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -243,7 +243,6 @@ def test_using_level(model):
"AutoLGBM",
"AutoNHITS",
"AutoTFT",
"PatchTST-FM",
]:
# These models do not support levels yet
with pytest.raises(ValueError) as excinfo:
Expand Down
5 changes: 0 additions & 5 deletions timecopilot/models/foundation/patchtst_fm.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -226,11 +226,6 @@ def forecast(
identifiers as the input DataFrame.
"""
freq = self._maybe_infer_freq(df, freq)
# When support for levels is added remove PatchTST-FM
# from the list of models that throw this exception in
# tests/models/test_models:test_using_level()
if level is not None:
raise ValueError("Level is not supported for patchtst-fm yet.")
qc = QuantileConverter(level=level, quantiles=quantiles)
dataset = TimeSeriesDataset.from_df(
df,
Expand Down
Loading