change raw to metadata (#75)

castorini · Sep 8, 2020 · 94befbd · 94befbd · justinborromeo · Sep 8, 2020
1 parent f692da6
commit 94befbd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pygaggle/model/writer.py b/pygaggle/model/writer.py
@@ -28,4 +28,4 @@ def write(self, scores: List[float], example: RelevanceExample):
         doc_scores = sorted(list(zip(example.documents, scores)),
                             key=lambda x: x[1], reverse=True)
         for ct, (doc, score) in enumerate(doc_scores):
-            self.write_line(f"{example.query.id}\t{doc.raw['docid']}\t{ct+1}")
+            self.write_line(f"{example.query.id}\t{doc.metadata['docid']}\t{ct+1}")