Fixed bug introduced in last commit.
@@ -269,6 +269,7 @@ class TrainingJob1toN(TrainingJob):
keys = self.train_po_keys
offsets = self.train_po_offsets
values = self.train_po_values
+
pairs[batch_index,] = keys[example_index]
start = offsets[example_index]
end = offsets[example_index + 1]
