Merge branch 'dev' into mamba

src/gluonts/torch/__init__.py

@@ -16,6 +16,7 @@
     "PyTorchPredictor",
     "DeepNPTSEstimator",
     "DeepAREstimator",
+    "TiDEEstimator",
     "SimpleFeedForwardEstimator",
     "TemporalFusionTransformerEstimator",
     "WaveNetEstimator",
@@ -28,6 +29,7 @@
 from .model.predictor import PyTorchPredictor
 from .model.deep_npts import DeepNPTSEstimator
 from .model.deepar import DeepAREstimator
+from .model.tide import TiDEEstimator
 from .model.simple_feedforward import SimpleFeedForwardEstimator
 from .model.tft import TemporalFusionTransformerEstimator
 from .model.wavenet import WaveNetEstimator

src/gluonts/torch/model/tide/__init__.py

@@ -0,0 +1,18 @@
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License").
+# You may not use this file except in compliance with the License.
+# A copy of the License is located at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the "license" file accompanying this file. This file is distributed
+# on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+# express or implied. See the License for the specific language governing
+# permissions and limitations under the License.
+
+from .module import TiDEModel
+from .lightning_module import TiDELightningModule
+from .estimator import TiDEEstimator
+
+__all__ = ["TiDEModel", "TiDELightningModule", "TiDEEstimator"]