Missing head in answer json added!

LiberAI · Jun 29, 2020 · eff2be1 · eff2be1
1 parent b321e0a
commit eff2be1
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/gsoc/zheyuan/utility/benchmark/retrieve_answers.py b/gsoc/zheyuan/utility/benchmark/retrieve_answers.py
@@ -29,7 +29,9 @@ def retrieve(query):
         for td in rows.find_all("a"):
             for a in td:
                 answer_dict = {
-                    "results": {
+                    "head" : {
+                      "vars" : [ "uri" ]
+                    },"results": {
                         "bindings": [{
                             "uri": {
                                 "type": "uri",
@@ -45,8 +47,11 @@ def retrieve(query):
                 if len(pre) <= 50:
                     answer_dict = {
                         "results": {
+                            "head": {
+                                "vars": ["string"]
+                            },
                             "bindings": [{
-                                "uri": {
+                                "string": {
                                     "type": "literal",
                                     "value": pre
                                 }
@@ -70,7 +75,7 @@ def retrieve(query):
     # query = args.query
     answer_groups = []
     i = 1
-    with open("../output_decoded.txt", 'r') as lines:
+    with open("./output_decoded.txt", 'r') as lines:
          for line in lines:
              i+=1
              try: