From 33cf2402a6632902c9fac016050e9f876ffe6a48 Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@gmail.com>
Date: Thu, 31 Mar 2022 12:07:14 +0200
Subject: [PATCH] fix repeated_past_target

---
 tft/module.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tft/module.py b/tft/module.py
index 7547c1f..c3d35b7 100644
--- a/tft/module.py
+++ b/tft/module.py
@@ -583,8 +583,9 @@ class TFTModel(nn.Module):
             repeats=self.num_parallel_samples, dim=0
         )
 
-        repeated_past_target = target.repeat_interleave(
-            repeats=self.num_parallel_samples, dim=0
+        repeated_past_target = (
+            past_target.repeat_interleave(repeats=self.num_parallel_samples, dim=0)
+            / repeated_scale
         )
         repeated_past_selection = past_selection.repeat_interleave(
             repeats=self.num_parallel_samples, dim=0