def METHOD_NAME():
    parser = argparse.ArgumentParser()
    parser.add_argument("--source_file", type=str, help="Source file to sample length distribution.")
    parser.add_argument("--output_dir", type=str, help="Output directory.")
    parser.add_argument("--max_prefix_size", type=int, default=25_000,
                        help="Number of maximum prefixes to keep while generating games.")
    parser.add_argument("--num_games", type=int, default=400_000, help="Number of games.")
    args = parser.METHOD_NAME()
    assert (args.source_file is not None)
    assert (args.output_dir is not None)
    return args