fixed reader only prediction (#286)

cdqa-suite · Oct 23, 2019 · 376ae3a · 376ae3a
1 parent fc9f511
commit 376ae3a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cdqa/reader/bertqa_sklearn.py b/cdqa/reader/bertqa_sklearn.py
@@ -785,7 +785,7 @@ def write_predictions(
         best_dict["qas_id"] = example.qas_id
         best_dict["title"] = example.title
         best_dict["paragraph"] = example.paragraph
-        best_dict["retriever_score"] = example.retriever_score.item()
+        best_dict["retriever_score"] = float(example.retriever_score)
         best_dict["final_score"] = (1 - retriever_score_weight) * (
             best_dict["start_logit"] + best_dict["end_logit"]
         ) + retriever_score_weight * best_dict["retriever_score"]

diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@ def read(file):
 
 setup(
     name="cdqa",
-    version="1.3.2",
+    version="1.3.3",
     author="Félix MIKAELIAN, André FARIAS, Matyas AMROUCHE, Olivier SANS, Théo NAZON",
     description="An End-To-End Closed Domain Question Answering System",
     long_description=read("README.md"),