From 38f740a1d579b6d21226fae4f758e315cbfa1243 Mon Sep 17 00:00:00 2001
From: VictorSanh <victorsanh@gmail.com>
Date: Fri, 2 Nov 2018 11:29:45 -0400
Subject: [PATCH] Fix bug writing predictions in run_squad_pytorch

---
 run_squad_pytorch.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/run_squad_pytorch.py b/run_squad_pytorch.py
index 100909b821..626759a085 100644
--- a/run_squad_pytorch.py
+++ b/run_squad_pytorch.py
@@ -910,8 +910,8 @@ def main():
             #end_logits = [x.item() for x in end_logits]
             end_logits = [x.view(-1).detach().cpu().numpy() for x in end_logits]
             for idx, i in enumerate(unique_id):
-                s = start_logits[idx]
-                e = end_logits[idx]
+                s = [float(x) for x in start_logits[idx]]
+                e = [float(x) for x in end_logits[idx]]
                 all_results.append(
                     RawResult(
                         unique_id=i,