def METHOD_NAME(
    self,
    hyp: Hypothesis,
    cache: Dict[str, Any],
) -> Tuple[
    torch.Tensor,
    Union[
        Tuple[torch.Tensor, Optional[torch.Tensor]], List[Optional[torch.Tensor]]
    ],
    torch.Tensor,