Create output path directory if necessary

c473d7e0 · janEbert · 3d14707a · c473d7e0
Commit c473d7e0 authored May 08, 2023 by janEbert
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

main.py main.py +2 -0

No files found.
--- a/main.py
+++ b/main.py
@@ -2,6 +2,7 @@ import argparse
 import json
 import logging
 import fnmatch
+import os

 from lm_eval import tasks, evaluator

@@ -94,6 +95,7 @@ def main():
    print(dumped)

    if args.output_path:
+        os.makedirs(os.path.dirname(args.output_path), exist_ok=True)
        with open(args.output_path, "w") as f:
            f.write(dumped)