From 7a7ddc5997d1df8b7562b9cf333142859c69d1e8 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Wed, 29 Jan 2020 14:25:22 +0100 Subject: [PATCH] typos --- .../transformer_tempflow_estimator.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pts/model/transformer_tempflow/transformer_tempflow_estimator.py b/pts/model/transformer_tempflow/transformer_tempflow_estimator.py index e6ab16c..aabef6b 100644 --- a/pts/model/transformer_tempflow/transformer_tempflow_estimator.py +++ b/pts/model/transformer_tempflow/transformer_tempflow_estimator.py @@ -163,8 +163,8 @@ class TransformerTempFlowEstimator(PTSEstimator): ] ) - def create_training_network(self, device: torch.device) -> TempFlowTrainingNetwork: - return TempFlowTrainingNetwork( + def create_training_network(self, device: torch.device) -> TransformerTempFlowTrainingNetwork: + return TransformerTempFlowTrainingNetwork( input_size=self.input_size, target_dim=self.target_dim, num_heads=self.num_heads, @@ -192,10 +192,10 @@ class TransformerTempFlowEstimator(PTSEstimator): def create_predictor( self, transformation: Transformation, - trained_network: TempFlowTrainingNetwork, + trained_network: TransformerTempFlowTrainingNetwork, device: torch.device, ) -> PTSPredictor: - prediction_network = TempFlowPredictionNetwork( + prediction_network = TransformerTempFlowPredictionNetwork( input_size=self.input_size, target_dim=self.target_dim, num_heads=self.num_heads,