1 files changed, 4 insertions, 3 deletions
diff --git a/model.py b/model.py
index 842c727..fd99168 100644
--- a/model.py
+++ b/model.py
@@ -367,9 +367,10 @@ class PrimedGenerator(Runner):
 
             for j in range(n):
                 ratio = i / (n-1)
-                a = (1-ratio) * sub_sequence_a[:, j].float()
-                b = ratio * sub_sequence_b[:, j].float()
-                tmp_sub_sequence[:, j] = ( a + b ).long()
+                for k in range(n_seqs):
+                    a = (1-ratio) * sub_sequence_a[k, j].float()
+                    b = ratio * sub_sequence_b[k, j].float()
+                    tmp_sub_sequence[k, j] = int( a + b )
 
             return tmp_sub_sequence