Improve base_model docs

sktime · Jun 4, 2021 · d6175c6 · d6175c6
1 parent c7b3933
commit d6175c6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytorch_forecasting/models/base_model.py b/pytorch_forecasting/models/base_model.py
@@ -159,8 +159,8 @@ def __init__(self, my_first_parameter: int=2, loss=SMAPE()):
                     super().__init__(loss=loss)
 
                 def forward(self, x):
-                    encoding_target = x["encoder_target"]
-                    prediction = self.transform_output(prediction=..., target_scale=x["target_scale"])
+                    normalized_prediction = self.module(x)
+                    prediction = self.transform_output(prediction=normalized_prediction, target_scale=x["target_scale"])
                     return self.to_network_output(prediction=prediction)
 
     """