if __name__ == "__main__":
args = parse_args()
generate(args)
After Change
if __name__ == "__main__":
args = parse_args()
if args.fold is not None:
folds = [args.fold]
else:
folds = c.BUZZER_GENERATION_FOLDS
for fold in folds:
log.info("Generating {} outputs".format(fold))
generate(fold, args.config)
