def METHOD_NAME(self, subset: Subset, **kwargs) -> Union[DataLoader, None]:
    """Loads subset (train/validation) partition of as pytorch DataLoader.
    Args:
        subset: Subset as an instance of PyTorch's `Subset`
    Returns:
        Data loader for `subset`. `None` if the subset is empty
    """
    # Return None if subset has no data
    if len(subset) <= 0:
        return None
    return self._create_torch_data_loader(subset, **kwargs)