def METHOD_NAME(self, source, target):
    if self.add_bos_token:
        source = torch.cat([source.new([self.vocab.bos()]), source])
        if target is not None:
            target = torch.cat([target.new([self.tgt_vocab.bos()]), target])
    return source, target