From d95bcfe57f9c967b852b1a67ac2d61f4d26c07d4 Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@gmail.com>
Date: Mon, 27 Jan 2020 20:15:03 +0100
Subject: [PATCH] fix imports

---
 pts/model/transformer/transformer_estimator.py | 10 ++++++----
 pts/model/transformer/transformer_network.py   |  7 +++++--
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/pts/model/transformer/transformer_estimator.py b/pts/model/transformer/transformer_estimator.py
index ebfaf8f..46d165e 100644
--- a/pts/model/transformer/transformer_estimator.py
+++ b/pts/model/transformer/transformer_estimator.py
@@ -6,8 +6,8 @@ import torch
 import torch.nn as nn
 
 from pts import Trainer
-from pts.model import PTSEstimator, PTSPredictor, copy_parameters
-from pts.modules import RealNVP
+from pts.model import PTSEstimator, Predictor, PTSPredictor, copy_parameters
+from pts.modules import DistributionOutput, StudentTOutput
 from pts.dataset import FieldName
 from pts.transform import (
     Transformation,
@@ -17,6 +17,8 @@ from pts.transform import (
     CDFtoGaussianTransform,
     cdf_to_gaussian_forward_transform,
     RenameFields,
+    RemoveFields,
+    AddAgeFeature,
     AsNumpyArray,
     ExpandDimArray,
     AddObservedValuesIndicator,
@@ -27,8 +29,8 @@ from pts.transform import (
 )
 from pts.feature import (
     TimeFeature,
-    fourier_time_features_from_frequency_str,
-    get_fourier_lags_for_frequency,
+    time_features_from_frequency_str,
+    get_lags_for_frequency,
 )
 
 from .transformer_network import TransformerTrainingNetwork, TransformerPredictionNetwork
diff --git a/pts/model/transformer/transformer_network.py b/pts/model/transformer/transformer_network.py
index 75bc140..b3f7585 100644
--- a/pts/model/transformer/transformer_network.py
+++ b/pts/model/transformer/transformer_network.py
@@ -10,8 +10,11 @@ from pts.modules import DistributionOutput, MeanScaler, NOPScaler, FeatureEmbedd
 from pts.model import weighted_average
 
 
-from .trans_encoder import TransformerEncoder
-from .trans_decoder import TransformerDecoder
+def prod(xs):
+    p = 1
+    for x in xs:
+        p *= x
+    return p
 
 
 class TransformerNetwork(nn.Module):