def METHOD_NAME(self, dataset_group_arn: str) -> DatasetGroup:
    try:
        return self.dataset_groups[dataset_group_arn]
    except KeyError:
        raise ResourceNotFoundException("No resource found " + dataset_group_arn)