Commit 76a0f7ac authored by alat-rights's avatar alat-rights
Browse files

fix doctest failure

parent 8c33822b
Loading
Loading
Loading
Loading
+3 −1
Original line number Diff line number Diff line
@@ -22,10 +22,12 @@ class BertFeaturizer(Featurizer):
  >>> from transformers import BertTokenizerFast
  >>> tokenizer = BertTokenizerFast.from_pretrained("Rostlab/prot_bert", do_lower_case=False)
  >>> featurizer = BertFeaturizer(tokenizer)
  >>> feats = featurizer.featurize('D L I P [MASK] L V T', return_tensors="pt")
  >>> feats = featurizer.featurize('D L I P [MASK] L V T')

  Notes
  -----
  Examples are based on RostLab's ProtBert documentation.

  This class may contain code and/or documentation taken from the
  RobertaFeaturizer pull request (#2581), which have been moved here due to
  code restructuring.