make to_file accept simple file name again

icecube · Sep 5, 2024 · 94785ca · 94785ca
1 parent 7753dea
commit 94785ca
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pisa/utils/fileio.py b/pisa/utils/fileio.py
@@ -545,8 +545,9 @@ def to_file(obj, fname, fmt=None, overwrite=True, warn=True, **kwargs):
         ext = fmt.lower()
 
     dirname = os.path.dirname(fname)
-
-    os.makedirs(dirname, exist_ok=True)
+    if dirname != "":
+        # would fail with FileNotFoundError otherwise
+        mkdir(dirname, warn=warn)
 
     if ext in ZIP_EXTS or ext in XOR_EXTS:
         rootname, inner_ext = os.path.splitext(rootname)