Update utils.py

aangelopoulos · May 29, 2024 · d26c293 · d26c293
1 parent e6b3ad4
commit d26c293
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/utils.py b/examples/utils.py
@@ -1,3 +1,4 @@
+import os
 import numpy as np
 import matplotlib.patheffects as pe
 import matplotlib.pyplot as plt
@@ -223,6 +224,7 @@ def make_plots(
  axs[0].set_yticklabels([])
 
  plt.tight_layout()
+ os.makedirs("/".join(plot_savename.split("/")[:-1]), exist_ok=True)
  plt.savefig(plot_savename)