def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('-d', '--dir', type=str, required=True, help='name of the data directory')
    args = parser.parse_args()
    return args.dir
def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('-d', '--dir', type=str, required=True, help='name of the data directory')
    args = parser.parse_args()
    return args.dir