for t in grids.all_tasks():
# for t in [grids.task_replace_color]:
print(t.__name__)
prompts, answers = grids.generate_prompts_and_answers_(nb, tasks=[t])
for t in grids.all_tasks():
# for t in [grids.task_replace_color]:
print(t.__name__)
prompts, answers = grids.generate_prompts_and_answers_(nb, tasks=[t])