[MNT] move model base classes into models.base module - part 1

pytorch_forecasting/models/__init__.py

@@ -2,7 +2,7 @@
 Models for timeseries forecasting.
 """
 
-from pytorch_forecasting.models.base_model import (
+from pytorch_forecasting.models.base import (
     AutoRegressiveBaseModel,
     AutoRegressiveBaseModelWithCovariates,
     BaseModel,

pytorch_forecasting/models/_base_model_rename_this.py

@@ -0,0 +1,17 @@
+"""Base classes for pytorch-foercasting models."""
+
+from pytorch_forecasting.models.base import (
+    AutoRegressiveBaseModel,
+    AutoRegressiveBaseModelWithCovariates,
+    BaseModel,
+    BaseModelWithCovariates,
+    Prediction,
+)
+
+__all__ = [
+    "AutoRegressiveBaseModel",
+    "AutoRegressiveBaseModelWithCovariates",
+    "BaseModel",
+    "BaseModelWithCovariates",
+    "Prediction",
+]

pytorch_forecasting/models/base/__init__.py

@@ -0,0 +1,17 @@
+"""Base classes for pytorch-foercasting models."""
+
+from pytorch_forecasting.models.base._base_model import (
+    AutoRegressiveBaseModel,
+    AutoRegressiveBaseModelWithCovariates,
+    BaseModel,
+    BaseModelWithCovariates,
+    Prediction,
+)
+
+__all__ = [
+    "AutoRegressiveBaseModel",
+    "AutoRegressiveBaseModelWithCovariates",
+    "BaseModel",
+    "BaseModelWithCovariates",
+    "Prediction",
+]

pytorch_forecasting/models/deepar/_deepar.py

@@ -26,7 +26,7 @@
     MultivariateDistributionLoss,
     NormalDistributionLoss,
 )
-from pytorch_forecasting.models.base_model import (
+from pytorch_forecasting.models.base import (
     AutoRegressiveBaseModelWithCovariates,
     Prediction,
 )

pytorch_forecasting/models/mlp/_decodermlp.py

@@ -18,7 +18,7 @@
     MultiHorizonMetric,
     QuantileLoss,
 )
-from pytorch_forecasting.models.base_model import BaseModelWithCovariates
+from pytorch_forecasting.models.base import BaseModelWithCovariates
 from pytorch_forecasting.models.mlp.submodules import FullyConnectedModule
 from pytorch_forecasting.models.nn.embeddings import MultiEmbedding
 

pytorch_forecasting/models/nbeats/_nbeats.py

@@ -10,7 +10,7 @@
 from pytorch_forecasting.data import TimeSeriesDataSet
 from pytorch_forecasting.data.encoders import NaNLabelEncoder
 from pytorch_forecasting.metrics import MAE, MAPE, MASE, RMSE, SMAPE, MultiHorizonMetric
-from pytorch_forecasting.models.base_model import BaseModel
+from pytorch_forecasting.models.base import BaseModel
 from pytorch_forecasting.models.nbeats.sub_modules import (
     NBEATSGenericBlock,
     NBEATSSeasonalBlock,

pytorch_forecasting/models/nhits/_nhits.py

@@ -20,7 +20,7 @@
     MultiHorizonMetric,
     MultiLoss,
 )
-from pytorch_forecasting.models.base_model import BaseModelWithCovariates
+from pytorch_forecasting.models.base import BaseModelWithCovariates
 from pytorch_forecasting.models.nhits.sub_modules import NHiTS as NHiTSModule
 from pytorch_forecasting.models.nn.embeddings import MultiEmbedding
 from pytorch_forecasting.utils import create_mask, to_list

pytorch_forecasting/models/rnn/_rnn.py

@@ -21,7 +21,7 @@
     MultiLoss,
     QuantileLoss,
 )
-from pytorch_forecasting.models.base_model import AutoRegressiveBaseModelWithCovariates
+from pytorch_forecasting.models.base import AutoRegressiveBaseModelWithCovariates
 from pytorch_forecasting.models.nn import HiddenState, MultiEmbedding, get_rnn
 from pytorch_forecasting.utils import apply_to_list, to_list
 

pytorch_forecasting/models/temporal_fusion_transformer/_tft.py

@@ -19,7 +19,7 @@
     MultiHorizonMetric,
     QuantileLoss,
 )
-from pytorch_forecasting.models.base_model import BaseModelWithCovariates
+from pytorch_forecasting.models.base import BaseModelWithCovariates
 from pytorch_forecasting.models.nn import LSTM, MultiEmbedding
 from pytorch_forecasting.models.temporal_fusion_transformer.sub_modules import (
     AddNorm,

pytorch_forecasting/models/tide/_tide.py

@@ -12,7 +12,7 @@
 from pytorch_forecasting.data import TimeSeriesDataSet
 from pytorch_forecasting.data.encoders import NaNLabelEncoder
 from pytorch_forecasting.metrics import MAE, MAPE, MASE, RMSE, SMAPE
-from pytorch_forecasting.models.base_model import BaseModelWithCovariates
+from pytorch_forecasting.models.base import BaseModelWithCovariates
 from pytorch_forecasting.models.nn.embeddings import MultiEmbedding
 from pytorch_forecasting.models.tide.sub_modules import _TideModule