Loading deepchem/feat/molecule_featurizers/roberta_tokenizer.py +1 −5 Original line number Diff line number Diff line Loading @@ -35,13 +35,9 @@ 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) #super().__init__(input_ids, attention_mask) self.input_ids = input_ids self.attention_mask = attention_mask return def _featurize(self, mol: RDKitMol) -> List[List[int]]: Loading Loading
deepchem/feat/molecule_featurizers/roberta_tokenizer.py +1 −5 Original line number Diff line number Diff line Loading @@ -35,13 +35,9 @@ 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) #super().__init__(input_ids, attention_mask) self.input_ids = input_ids self.attention_mask = attention_mask return def _featurize(self, mol: RDKitMol) -> List[List[int]]: Loading
