005e5a8f0b7b6156d6e0e0b0587a07479185c971,bugbug/labels.py,,,#,83

Before Change

if __name__ == "__main__":
    classes = get_bugbug_labels(augmentation=False)
    bugzilla.download_and_store_bugs([bug_id for bug_id in classes.keys()])

After Change

if __name__ == "__main__":
    bug_ids = set()
    for csv_file in os.listdir("labels"):
        with open(os.path.join("labels", csv_file)) as f:
            reader = csv.reader(f)
            // Assume the first row is the header.
            next(reader)
            // Assume the first column is the bug ID.
            bug_ids.update([row[0] for row in reader])

    bugzilla.download_bugs(bug_ids)

Instances