It would be good to update the argument description to mention that it also takes a dataset name like in get_datasets. I did not notice it the first time I looked around.