Fix output dir creation

william-galvin · Dec 27, 2023 · f7942a8 · f7942a8
1 parent 228a9e0
commit f7942a8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pennylane/vqe_pennylane/main.py b/pennylane/vqe_pennylane/main.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 
 import argparse
+import os
 
 from pennylane import numpy as np
 import numpy
@@ -108,6 +109,9 @@ def cost_fn(param):
     plt.xticks(fontsize=12)
     plt.yticks(fontsize=12)
 
+    if not os.path.exists(args.output_dir):
+        os.makedirs(args.output_dir)
+
     plt.savefig(f"{args.output_dir}/plot.png")
     print(f"\nPlot saved at {args.output_dir}/plot.png")