Skip to content

Commit 18153aa

Browse files
authored
[ENH] move tide model - part 2 (#1744)
Follows #1743, adds exports and linting.
1 parent 8bcc0d8 commit 18153aa

File tree

2 files changed

+15
-6
lines changed

2 files changed

+15
-6
lines changed
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
"""Tide model."""
2+
3+
from pytorch_forecasting.models.tide._tide import TiDEModel
4+
from pytorch_forecasting.models.tide.sub_modules import _TideModule
5+
6+
__all__ = [
7+
"_TideModule",
8+
"TiDEModel",
9+
]

pytorch_forecasting/models/tide/_tide.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,15 @@
1-
from typing import Dict, List, Optional, Tuple, Union
21
from copy import copy
2+
from typing import Dict, List, Optional, Tuple, Union
3+
4+
import torch
5+
from torch import nn
36

7+
from pytorch_forecasting.data import TimeSeriesDataSet
8+
from pytorch_forecasting.data.encoders import NaNLabelEncoder
49
from pytorch_forecasting.metrics import MAE, MAPE, MASE, RMSE, SMAPE
510
from pytorch_forecasting.models.base_model import BaseModelWithCovariates
611
from pytorch_forecasting.models.nn.embeddings import MultiEmbedding
712
from pytorch_forecasting.models.tide.sub_modules import _TideModule
8-
from pytorch_forecasting.data import TimeSeriesDataSet
9-
from pytorch_forecasting.data.encoders import NaNLabelEncoder
10-
11-
from torch import nn
12-
import torch
1313

1414

1515
class TiDEModel(BaseModelWithCovariates):

0 commit comments

Comments
 (0)