if __name__ == "__main__":
classes = get_bugbug_labels(augmentation=False)
bugzilla.download_and_store_bugs([bug_id for bug_id in classes.keys()])
After Change
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)
