Add creation of direcotry, if it doesnt exist

Roxana-P · Sep 28, 2023 · b731d4f · b731d4f
1 parent 0cc563b
commit b731d4f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/examples/export_npz.py b/examples/export_npz.py
@@ -1,3 +1,5 @@
+from pathlib import Path
+
 import numpy as np
 
 import gustaf
@@ -54,6 +56,7 @@
     hexa.show_options["data_name"] = "arange"
 
     # Export hexa, edges and vertices
+    Path("export").mkdir(exist_ok=True)
     gustaf.io.npz.export(hexa, "export/hexa.npz")
     gustaf.io.npz.export(hexa.to_edges(), "export/hexa_edges.npz")
     gustaf.io.npz.export(hexa.to_vertices(), "export/hexa_vertices.npz")