def METHOD_NAME(text: str,
                      tokenizer: RobertaTokenizer) -> Tuple[str, torch.Tensor]:
  tokenized_sentence = tokenizer.encode(text, add_special_tokens=True)
  # Workaround to manually remove batch dim until we have the feature to
  # add optional batching flag.
  # TODO(https://github.com/apache/beam/issues/21863): Remove once optional
  # batching flag added
  return text, torch.tensor(tokenized_sentence).numpy()