def METHOD_NAME():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-d",
        "--datasets",
        nargs="+",
        required=True,
        help="""
        Multiple datasets can be passed to set different options.
        For example, run as:
           ./check_dataset_counts.py --datasets math oasst_export_eu
        to check the counts of the math and the oasst_export_eu dataset.
    """,
    )
    parser.add_argument("--mode", dest="mode", type=Mode, choices=list(Mode))
    parser.add_argument("--cache_dir", dest="cache_dir", type=str)
    parser.add_argument("--verbose", dest="verbose", type=str, default=False)
    args, _ = parser.parse_known_args()
    return args