Fix make_profile_plots args for not-main case

python/MooreTests/make_profile_plots.py

@@ -149,10 +149,16 @@ def main(listOfLogs, hltlabel, throughput, produceYAML, ht_file, heaptrack,
             'facecolor': 'white',
             'edgecolor': 'none'
         })
-    if args.output_dir and args.output_dir[-1] != "/":
-        args.output_dir = args.output_dir + "/"
-    plt.savefig(f"{args.output_dir}FlameBars.png", bbox_inches='tight')
-    plt.savefig(f"{args.output_dir}FlameBars.pdf", bbox_inches='tight')
+
+    try:
+        # this only works in main
+        if args.output_dir and args.output_dir[-1] != "/":
+            args.output_dir = args.output_dir + "/"
+        plt.savefig(f"{args.output_dir}FlameBars.png", bbox_inches='tight')
+        plt.savefig(f"{args.output_dir}FlameBars.pdf", bbox_inches='tight')
+    except NameError:
+        plt.savefig("FlameBars.png", bbox_inches='tight')
+        plt.savefig("FlameBars.pdf", bbox_inches='tight')
 
 
 if __name__ == '__main__':