Skip to content

Commit 04d5d96

Browse files
feat: shorten package import path from pytorch_forecasting to ptf with legacy redirect
1 parent 4757357 commit 04d5d96

205 files changed

Lines changed: 815 additions & 839 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

.gitignore

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,4 +146,4 @@ lightning_logs
146146
.DS_Store
147147

148148
# data
149-
pytorch_forecasting/data/*.parquet
149+
ptf/data/*.parquet

ptf/__init__.py

Lines changed: 18 additions & 118 deletions
Original file line numberDiff line numberDiff line change
@@ -1,121 +1,21 @@
1-
"""
2-
PyTorch Forecasting package for timeseries forecasting with PyTorch.
3-
"""
1+
import sys
2+
from importlib.abc import MetaPathFinder
43

5-
__version__ = "1.8.0"
4+
# Custom import interceptor to redirect all sub-imports (e.g., ptf.models -> pytorch_forecasting.models)
5+
class LegacyRedirectFinder(MetaPathFinder):
6+
def find_spec(self, fullname, path, target=None):
7+
if fullname.startswith("ptf"):
8+
real_name = fullname.replace("ptf", "pytorch_forecasting", 1)
9+
try:
10+
__import__(real_name)
11+
sys.modules[fullname] = sys.modules[real_name]
12+
return sys.modules[real_name].__spec__
13+
except ImportError:
14+
return None
15+
return None
616

7-
from ptf.data import (
8-
EncoderNormalizer,
9-
GroupNormalizer,
10-
MultiNormalizer,
11-
NaNLabelEncoder,
12-
TimeSeriesDataSet,
13-
)
14-
from ptf.metrics import (
15-
MAE,
16-
MAPE,
17-
MASE,
18-
RMSE,
19-
SMAPE,
20-
BetaDistributionLoss,
21-
CrossEntropy,
22-
DistributionLoss,
23-
ImplicitQuantileNetworkDistributionLoss,
24-
LogNormalDistributionLoss,
25-
MQF2DistributionLoss,
26-
MultiHorizonMetric,
27-
MultiLoss,
28-
MultivariateNormalDistributionLoss,
29-
NegativeBinomialDistributionLoss,
30-
NormalDistributionLoss,
31-
PoissonLoss,
32-
QuantileLoss,
33-
)
34-
from ptf.models import (
35-
GRU,
36-
LSTM,
37-
AutoRegressiveBaseModel,
38-
AutoRegressiveBaseModelWithCovariates,
39-
Baseline,
40-
BaseModel,
41-
BaseModelWithCovariates,
42-
DecoderMLP,
43-
DeepAR,
44-
MultiEmbedding,
45-
NBeats,
46-
NBeatsKAN,
47-
NHiTS,
48-
RecurrentNetwork,
49-
TemporalFusionTransformer,
50-
TiDEModel,
51-
get_rnn,
52-
)
53-
from ptf.utils import (
54-
apply_to_list,
55-
autocorrelation,
56-
create_mask,
57-
detach,
58-
get_embedding_size,
59-
groupby_apply,
60-
integer_histogram,
61-
move_to_device,
62-
profile,
63-
to_list,
64-
unpack_sequence,
65-
)
66-
from ptf.utils._maint._show_versions import show_versions
17+
sys.meta_path.insert(0, LegacyRedirectFinder())
6718

68-
__all__ = [
69-
"TimeSeriesDataSet",
70-
"GroupNormalizer",
71-
"EncoderNormalizer",
72-
"NaNLabelEncoder",
73-
"MultiNormalizer",
74-
"TemporalFusionTransformer",
75-
"TiDEModel",
76-
"NBeats",
77-
"NBeatsKAN",
78-
"NHiTS",
79-
"Baseline",
80-
"DeepAR",
81-
"BaseModel",
82-
"BaseModelWithCovariates",
83-
"AutoRegressiveBaseModel",
84-
"AutoRegressiveBaseModelWithCovariates",
85-
"MultiHorizonMetric",
86-
"MultiLoss",
87-
"MAE",
88-
"MAPE",
89-
"MASE",
90-
"SMAPE",
91-
"DistributionLoss",
92-
"BetaDistributionLoss",
93-
"LogNormalDistributionLoss",
94-
"NegativeBinomialDistributionLoss",
95-
"NormalDistributionLoss",
96-
"ImplicitQuantileNetworkDistributionLoss",
97-
"MultivariateNormalDistributionLoss",
98-
"MQF2DistributionLoss",
99-
"CrossEntropy",
100-
"PoissonLoss",
101-
"QuantileLoss",
102-
"RMSE",
103-
"get_rnn",
104-
"LSTM",
105-
"GRU",
106-
"MultiEmbedding",
107-
"apply_to_list",
108-
"autocorrelation",
109-
"get_embedding_size",
110-
"create_mask",
111-
"to_list",
112-
"RecurrentNetwork",
113-
"DecoderMLP",
114-
"detach",
115-
"move_to_device",
116-
"integer_histogram",
117-
"groupby_apply",
118-
"profile",
119-
"show_versions",
120-
"unpack_sequence",
121-
]
19+
# Map the root package to pytorch_forecasting
20+
import pytorch_forecasting
21+
sys.modules["ptf"] = pytorch_forecasting

ptf/data/tests/__init__.py

Lines changed: 0 additions & 1 deletion
This file was deleted.

ptf/layers/_blocks/__init__.py

Lines changed: 0 additions & 3 deletions
This file was deleted.

ptf/layers/_embeddings/__init__.py

Lines changed: 0 additions & 19 deletions
This file was deleted.

ptf/layers/_encoders/__init__.py

Lines changed: 0 additions & 8 deletions
This file was deleted.

ptf/layers/_recurrent/_mlstm/__init__.py

Lines changed: 0 additions & 7 deletions
This file was deleted.

ptf/layers/_recurrent/_slstm/__init__.py

Lines changed: 0 additions & 7 deletions
This file was deleted.

ptf/metrics/_distributions_pkg/__init__.py

Lines changed: 0 additions & 35 deletions
This file was deleted.

ptf/metrics/_point_pkg/__init__.py

Lines changed: 0 additions & 29 deletions
This file was deleted.

0 commit comments

Comments
 (0)