Change to use a `dataset.subset(groups_dict)`.
Change to use a
dataset.subset(groups_dict).