expose smiles tokenizer under import guard (fe513da2) · Commits · 钟慕尧 / deepchem

deepchem/feat/init.py

+10 −0

Original line number	Diff line number	Diff line
		@@ -25,3 +25,13 @@ from deepchem.feat.smiles_featurizers import SmilesToSeq, SmilesToImage
		from deepchem.feat.material_featurizers import ElementPropertyFingerprint
		from deepchem.feat.material_featurizers import SineCoulombMatrix
		from deepchem.feat.material_featurizers import CGCNNFeaturizer

		try:
		import transformers
		from transformers import BertTokenizer

		from deepchem.feat.smiles_tokenizer import SmilesTokenizer
		from deepchem.feat.smiles_tokenizer import BasicSmilesTokenizer
		except ImportError:
		logger.warning("HuggingFace transformers is not available. Please install using 'pip install transformers' to use the SmilesTokenizer")