Update run_hf.py

raishish · Feb 23, 2024 · 6653f72 · 6653f72
1 parent d687ca4
commit 6653f72
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/model/run_hf.py b/model/run_hf.py
@@ -251,8 +251,8 @@ def run_inference(dataset_names, dataset_split, hf_dataset_name, prompt_types, m
             else:
                 final_res_dir = f'{results_dir}/{pt.value}.pkl'
 
-            if not os.path.exists(final_res_dir):
-                os.makedirs(final_res_dir)
+            if not os.path.exists('/'.join(final_res_dir.split('/')[:-1])):
+                os.makedirs('/'.join(final_res_dir.split('/')[:-1]))
             with open(final_res_dir, 'wb') as handle:
                 pickle.dump(answers, handle, protocol=pickle.HIGHEST_PROTOCOL)