fix the scaler

Author: kashif

pts/model/causal_deepar/causal_deepar_network.py

@@ -177,8 +177,8 @@ def unroll_encoder(
         # scale is computed on the context length last units of the past target
         # scale shape is (batch_size, 1, *target_shape)
         _, scale = self.scaler(
-            past_target[:, self.context_length :, ...],
-            past_observed_values[:, self.context_length :, ...],
+            past_target[:, -self.context_length :, ...],
+            past_observed_values[:, -self.context_length :, ...],
         )
 
         _, control_scale = self.control_scaler(

pts/model/deepar/deepar_network.py

@@ -160,8 +160,8 @@ def unroll_encoder(
         # scale is computed on the context length last units of the past target
         # scale shape is (batch_size, 1, *target_shape)
         _, scale = self.scaler(
-            past_target[:, self.context_length :, ...],
-            past_observed_values[:, self.context_length :, ...],
+            past_target[:, -self.context_length :, ...],
+            past_observed_values[:, -self.context_length :, ...],
         )
 
         # (batch_size, num_features)