yapf, flake8 (f9bf60f5) · Commits · 钟慕尧 / deepchem

deepchem/feat/molecule_featurizers/roberta_tokenizer.py

+3 −4

Original line number	Diff line number	Diff line
		@@ -35,7 +35,6 @@ class RobertaFeaturizer(RobertaTokenizerFast, MolecularFeaturizer):
		as well as DeepChem's MolecularFeaturizer class.
		"""


		#def __init__(self, **kwargs):
		def __init__(self, input_ids, attention_mask):
		# super().__init__(**kwargs)
		@@ -45,7 +44,6 @@ class RobertaFeaturizer(RobertaTokenizerFast, MolecularFeaturizer):

		return


		def _featurize(self, mol: RDKitMol) -> List[List[int]]:
		"""Calculate encoding using HuggingFace's RobertaTokenizerFast

		@@ -68,7 +66,8 @@ class RobertaFeaturizer(RobertaTokenizerFast, MolecularFeaturizer):
		# the encoding is natively a dictionary with keys 'input_ids' and 'attention_mask'
		# -> make this a list of two lists to allow np to handle it
		# encoding = list(self(smiles_string, **kwargs).values())
		encoding = list(self(smiles_string, self.input_ids, self.attention_mask).values())
		encoding = list(
		self(smiles_string, self.input_ids, self.attention_mask).values())

		return encoding