diff --git a/python/examples/shower_profile.py b/python/examples/shower_profile.py index 7be09e4236ebc3b122f1b57a95839f374f3c9c3d..340afa201eb94be50ab4e3ff0a82730fde0251c9 100644 --- a/python/examples/shower_profile.py +++ b/python/examples/shower_profile.py @@ -78,6 +78,7 @@ def draw_profiles(pr, pr_config): ax.set_xlim(min(pr["X"]), max(pr["X"])) + plot_path = os.path.join(args.output_dir, "shower_profile_long_profiles.png") print("Saving", plot_path) fig.savefig(plot_path) @@ -94,6 +95,7 @@ def draw_energyloss(prdx, prdx_config): ax.set_xlim(min(prdx["X"]), max(prdx["X"])) + plot_path = os.path.join(args.output_dir, "shower_profile_energy_loss.png") print("Saving", plot_path) fig.savefig(plot_path)