Update data_utils.py

data_utils.py

@@ -120,6 +120,7 @@ def sample_an_eval_result(eval_results, model_list=[]):
     random.shuffle(eval_results)
     
     for eval_item in eval_results:
+        print(eval_item.keys())
         model = eval_item['model']
         task_type = eval_item['task_type'] # primary task type
         if model not in model_list:
@@ -135,6 +136,9 @@ def sample_an_eval_result(eval_results, model_list=[]):
                 "task_type": task_type,
                 "plan_history": plan_history,
                 "ground_history": ground_history,
+                "pred": eval_item['pred'],
+                "answer": eval_item['answer'],
+                "correctness": eval_item['correctness'],
                 "image": eval_item['image'].replace("eval/aokvqa/images/val2017/", "file/data_dir/test_images/")
             }
         else:
@@ -144,6 +148,9 @@ def sample_an_eval_result(eval_results, model_list=[]):
                 "task_type": task_type,
                 "plan_history": plan_history,
                 "ground_history": ground_history,
+                "pred": eval_item['pred'],
+                "answer": eval_item['answer'],
+                "correctness": eval_item['correctness'],
                 "image": None
             }
         break