da35644f07a8c970d4f571c1ecb94bde3065d8de,oneoffs/oneoff_utils.py,,find_and_filter_sgf_files,#Any#Any#Any#,68

Before Change

    print("Finding all sgf files in {} with year >= {} and komi = {}".format(
        base_dir, min_year, komi))
    count = 0
    for dirpath, dirnames, filenames in tqdm(os.walk(base_dir)):
        for filename in filenames:
            count += 1
            if count % 5000 == 0:

After Change

        return sgf_files

    f = filter_year_komi(min_year, komi)
    filtered_sgf_files = [sgf for sgf in tqdm(sgf_files) if f(sgf)]

    print("{} of {} .sgf files matched (min_year >= {}, komi = {})".format(
        len(filtered_sgf_files), len(sgf_files), min_year, komi))
    return filtered_sgf_files

Instances