Commit 16766840 authored by seyonechithrananda's avatar seyonechithrananda
Browse files

move import check to init

parent 71015d25
Loading
Loading
Loading
Loading
+3 −0
Original line number Diff line number Diff line
@@ -65,12 +65,15 @@ from deepchem.feat.material_featurizers import LCNNFeaturizer
from deepchem.feat.atomic_conformation import AtomicConformation
from deepchem.feat.atomic_conformation import AtomicConformationFeaturizer

# tokenizers
try:
  import transformers
  from transformers import BertTokenizer
  from transformers import RobertaTokenizerFast

  from deepchem.feat.smiles_tokenizer import SmilesTokenizer
  from deepchem.feat.smiles_tokenizer import BasicSmilesTokenizer
  from deepchem.feat.roberta_tokenizer import RobertaFeaturizer
except ModuleNotFoundError:
  pass

+1 −5
Original line number Diff line number Diff line
from deepchem.feat import MolecularFeaturizer
try:
from transformers import RobertaTokenizerFast
except:
  raise Warning('This class requires transformers.')
  pass
from deepchem.utils.typing import RDKitMol

from typing import Dict, List