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

fix doctest failure

parent 8c33822b
......@@ -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')
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.
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment