about summary refs log tree commit diff
path: root/gnqa/paper2_eval/src/retrieve_context.py
diff options
context:
space:
mode:
Diffstat (limited to 'gnqa/paper2_eval/src/retrieve_context.py')
-rw-r--r--gnqa/paper2_eval/src/retrieve_context.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/gnqa/paper2_eval/src/retrieve_context.py b/gnqa/paper2_eval/src/retrieve_context.py
index 58b9d472..fca90dd9 100644
--- a/gnqa/paper2_eval/src/retrieve_context.py
+++ b/gnqa/paper2_eval/src/retrieve_context.py
@@ -65,11 +65,11 @@ def create_datasets(query_list, domain, level):
     time.sleep(10) # sleep a bit to not overtask the api
     if ndx % 5 == 0:
       print('Will print to file number {0}'.format(int(ndx/5)))
-      outp_file  = '{0}dataset_{1}_{2}_{3}.json'.format(config['out.response.dataset']['gpt4o_dir'],level,domain,str(int(ndx/5)))
+      outp_file  = '{0}dataset_{1}_{2}_{3}.json'.format(config['out.response.dataset']['human_dir'],level,domain,str(int(ndx/5)))
       writeDatasetFile(responses, outp_file)
       responses = reset_responses()
   if len(responses['question']) > 0:
-    outp_file  = '{0}dataset_{1}_{2}_{3}.json'.format(config['out.response.dataset']['gpt4o_dir'],level,domain,str(int(ndx/5)+1))
+    outp_file  = '{0}dataset_{1}_{2}_{3}.json'.format(config['out.response.dataset']['human_dir'],level,domain,str(int(ndx/5)+1))
     writeDatasetFile(responses, outp_file)
 
 def parse_responses(jsonfile):