Skip to content

[ENH] Add forecaster support and remove _pkg class for new API #2434

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
phoeenniixx wants to merge 6 commits into
base: dm
Choose a base branch
from
+2,768 −1,523
Open

[ENH] Add forecaster support and remove _pkg class for new API #2434

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
6f21533
add forecaster refactor
phoeenniixx Sep 20, 2026
1b934cc
updates
phoeenniixx Sep 28, 2026
7fbc169
solve checkpointing failures
phoeenniixx Sep 28, 2026
e7624d8
update workflow target branch
phoeenniixx Sep 28, 2026
5c5cec7
rename BasePtforecaster to BasePtObject, reimplement check_is_fitted …
phoeenniixx Oct 4, 2026
bbedfa8
rename pkg files and bring all v2 pkg to forecaster
phoeenniixx Oct 8, 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
4 changes: 2 additions & 2 deletions .github/workflows/test.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,9 @@ name: Test

on:
push:
branches: [v2-dev]
branches: [dm]

@phoeenniixx phoeenniixx Sep 28, 2026

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

need to change this once #2423 is merged!

pull_request:
branches: [v2-dev]
branches: [dm]

concurrency:
group: ${{ github.workflow }}-${{ github.ref }}
Expand Down
22 changes: 11 additions & 11 deletions docs/source/pkg_v2.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -94,14 +94,14 @@ See the detailed API documentation for the available V2 Package classes below:
.. autosummary::
:toctree: api

models.temporal_fusion_transformer._tft_pkg_v2.TFT_pkg_v2
models.dlinear._dlinear_pkg_v2.DLinear_pkg_v2
models.samformer._samformer_v2_pkg.Samformer_pkg_v2
models.tide._tide_dsipts._tide_v2_pkg.TIDE_pkg_v2
models.timexer._timexer_pkg_v2.TimeXer_pkg_v2
models.xlstm._xlstm_pkg_v2.xLSTMTime_pkg_v2
models.mlp._decodermlp_pkg_v2.DecoderMLP_pkg_v2
models.softs._softs_pkg_v2.SOFTS_pkg_v2
models.scinet._scinet_pkg_v2.SCINet_pkg_v2
models.patch_tst._patch_tst_pkg_v2.PatchTST_pkg_v2
models.frets._frets_pkg_v2.FreTS_pkg_v2
models.temporal_fusion_transformer._tft_forecaster_v2.TFTForecaster
models.dlinear._dlinear_forecaster_v2.DLinearForecaster
models.samformer._samformer_forecaster_v2.SamformerForecaster
models.tide._tide_dsipts._tide_forecaster_v2.TIDEForecaster
models.timexer._timexer_forecaster_v2.TimeXerForecaster
models.xlstm._xlstm_forecaster_v2.xLSTMTimeForecaster
models.mlp._decodermlp_forecaster_v2.DecoderMLPForecaster
models.softs._softs_forecaster_v2.SOFTSForecaster
models.scinet._scinet_forecaster_v2.SCINetForecaster
models.patch_tst._patch_tst_forecaster_v2.PatchTSTForecaster
models.frets._frets_forecaster_v2.FreTSForecaster
313 changes: 0 additions & 313 deletions pytorch_forecasting/base/_base_pkg.py
View file
Open in desktop

This file was deleted.

16 changes: 8 additions & 8 deletions pytorch_forecasting/models/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,20 +10,20 @@
)
from pytorch_forecasting.models.baseline import Baseline
from pytorch_forecasting.models.deepar import DeepAR
from pytorch_forecasting.models.frets import FreTS, FreTS_pkg_v2
from pytorch_forecasting.models.frets import FreTS, FreTSForecaster
from pytorch_forecasting.models.mlp import DecoderMLP
from pytorch_forecasting.models.nbeats import NBeats, NBeatsKAN
from pytorch_forecasting.models.nhits import NHiTS
from pytorch_forecasting.models.nn import GRU, LSTM, MultiEmbedding, get_rnn
from pytorch_forecasting.models.patch_tst import (
PatchTST,
PatchTST_pkg,
PatchTST_pkg_v2,
PatchTST_v2,
PatchTSTForecaster,
)
from pytorch_forecasting.models.rnn import RecurrentNetwork
from pytorch_forecasting.models.scinet import SCINet_pkg_v2, SCINet_v2
from pytorch_forecasting.models.softs import SOFTS, SOFTS_pkg_v2
from pytorch_forecasting.models.scinet import SCINet_v2, SCINetForecaster
from pytorch_forecasting.models.softs import SOFTS, SOFTSForecaster
from pytorch_forecasting.models.temporal_fusion_transformer import (
TemporalFusionTransformer,
)
Expand All @@ -38,7 +38,7 @@
"PatchTST",
"PatchTST_v2",
"PatchTST_pkg",
"PatchTST_pkg_v2",
"PatchTSTForecaster",
"TemporalFusionTransformer",
"RecurrentNetwork",
"DeepAR",
Expand All @@ -56,9 +56,9 @@
"TimeXer",
"xLSTMTime",
"SOFTS",
"SOFTS_pkg_v2",
"SOFTSForecaster",
"SCINet_v2",
"SCINet_pkg_v2",
"SCINetForecaster",
"FreTS",
"FreTS_pkg_v2",
"FreTSForecaster",
]
Loading
Loading