Create dir

castorini · Mar 14, 2021 · 5414bca · 5414bca
1 parent bdd9973
commit 5414bca
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pyserini/output_writer.py b/pyserini/output_writer.py
@@ -1,4 +1,5 @@
 import json
+import os
 
 from abc import ABC, abstractmethod
 from enum import Enum, unique
@@ -30,6 +31,9 @@ def __init__(self, file_path: str, mode: str = 'w',
         self.file = None
 
     def __enter__(self):
+        dirname = os.path.dirname(self.file_path)
+        if dirname:
+            os.makedirs(dirname, exist_ok=True)
         self.file = open(self.file_path, self.mode)
         return self