Minor change in plot filename

johnantonn · Mar 18, 2022 · 08ecc40 · 08ecc40
1 parent 777380c
commit 08ecc40
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/utils.py b/src/utils.py
@@ -380,4 +380,5 @@ def plot_performance(out_dirname, total_budget):
         labels, handles = zip(
             *sorted(zip(labels, handles), key=lambda t: t[0]))
         ax.legend(handles, labels)
-    plt.savefig(os.path.join(path, 'all.png'))
+    fig_title = 'all_' + str(total_budget) + '.png'
+    plt.savefig(os.path.join(path, fig_title))