1import os 2 3def save_fig(fig, fileName, options): 4 if options.output: 5 output_path = os.path.join(options.output, fileName) 6 else: 7 output_path = fileName 8 9 fig.savefig(output_path)