def METHOD_NAME(self, data: Tuple[str, Any]) -> Optional[int]:
    return {
        "annotations": OxfordIIITPetDemux.SPLIT_AND_CLASSIFICATION,
        "trimaps": OxfordIIITPetDemux.SEGMENTATIONS,
    }.get(pathlib.Path(data[0]).parent.name)