Maybe this could be done by a string index? Something like `dataset["<task_name>"]`?
Maybe this could be done by a string index? Something like
dataset["<task_name>"]?