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)