Commit 04d4fc4e authored by alat-rights's avatar alat-rights
Browse files

modified __call__

parent 28a9dd70
Loading
Loading
Loading
Loading
+4 −2
Original line number Diff line number Diff line
from deepchem.feat import Featurizer
import numpy as np
from icecream import ic
from typing import Dict, List
try:
  from transformers import BertTokenizerFast
@@ -55,5 +57,5 @@ class BertFeaturizer(Featurizer):
    encoding = list(self.tokenizer(datapoint, **kwargs).values())
    return encoding

  def __call__(self, *args, **kwargs) -> Dict[str, List[int]]:
    return self.tokenizer.__call__(*args, **kwargs)
  def __call__(self, *args, **kwargs) -> np.ndarray:
    return self.featurize(*args, **kwargs)