create outputdir of not exists

logikon-ai · Mar 17, 2024 · 3e76a7b · 3e76a7b
1 parent f1b2231
commit 3e76a7b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/create_lm_eval_harness_tasks.py b/scripts/create_lm_eval_harness_tasks.py
@@ -38,7 +38,8 @@ def main():
     if args.output_dir is None:
         raise ValueError("output_dir must be specified")
     if not os.path.isdir(args.output_dir):
-        raise ValueError("output_dir must be a directory")
+        logging.info("output_dir does not exist, creating it")
+        os.makedirs(args.output_dir)
     if not os.path.isdir(args.configs_dir):
         raise ValueError(f"configs_dir is not a directory: {args.configs_dir}")