def forward_loss(self, data_points: Union[(List[Sentence], Sentence)]) -> Union[(torch.Tensor, Tuple[(torch.Tensor, int)])]:
    if (not isinstance(data_points, list)):
        data_points = [data_points]
    sentences = self._get_tars_formatted_sentences(data_points)
    loss = self.tars_model.forward_loss(sentences)
    return loss
def forward_loss(self, data_points: Union[(List[Sentence], Sentence)]) -> Union[(torch.Tensor, Tuple[(torch.Tensor, int)])]:
    if (not isinstance(data_points, list)):
        data_points = [data_points]
    sentences = self._get_tars_formatted_sentences(data_points)
    loss = self.tars_model.forward_loss(sentences)
    return loss