def parse_args():
    parser = argparse.ArgumentParser(description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter)
    parser.add_argument('input_dir', type=str, help='Input directory')
    parser.add_argument('plot_filebase', type=str, help='Plots directory')
    parser.add_argument('tag', type=str, help='OP tag')
    parser.add_argument('--systems', nargs='+', type=str, help='Systems')
    parser.add_argument('--varis', nargs='+', type=str, help='Simulation variants')
    parser.add_argument('--post_lfes', nargs='+', type=str, help='Filename additions after lfes, if any')
    parser.add_argument('--stacking_enes', nargs='+', type=float, help='Stacking energies (for colormap)')
    return parser.parse_args()
def parse_args():
    parser = argparse.ArgumentParser(description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter)
    parser.add_argument('input_dir', type=str, help='Input directory')
    parser.add_argument('plot_filebase', type=str, help='Plots directory')
    parser.add_argument('tag', type=str, help='OP tag')
    parser.add_argument('--systems', nargs='+', type=str, help='Systems')
    parser.add_argument('--varis', nargs='+', type=str, help='Simulation variants')
    parser.add_argument('--post_lfes', nargs='+', type=str, help='Filename additions after lfes, if any')
    parser.add_argument('--stacking_enes', nargs='+', type=float, help='Stacking energies (for colormap)')
    return parser.parse_args()