minor refactor and yapf (0a29133a) · Commits · 钟慕尧 / deepchem

deepchem/molnet/load_function/hiv_datasets.py

+0 −1

Original line number	Diff line number	Diff line
		@@ -43,7 +43,6 @@ def load_hiv(featurizer='ECFP', split='index', reload=True, **kwargs):
		featurizer = deepchem.feat.RawFeaturizer()
		elif featurizer == "smiles2img":
		img_spec = kwargs.get("img_spec", "std")
		logger.info(img_spec)
		featurizer = deepchem.feat.SmilesToImage(img_spec=img_spec)

		loader = deepchem.data.CSVLoader(

+5 −2

Original line number	Diff line number	Diff line
		@@ -11,7 +11,11 @@ import deepchem
		logger = logging.getLogger(__name__)


		def load_sampl(featurizer='ECFP', split='index', reload=True, move_mean=True, **kwargs):
		def load_sampl(featurizer='ECFP',
		split='index',
		reload=True,
		move_mean=True,
		**kwargs):
		"""Load SAMPL datasets."""
		# Featurize SAMPL dataset
		logger.info("About to featurize SAMPL dataset.")
		@@ -48,7 +52,6 @@ def load_sampl(featurizer='ECFP', split='index', reload=True, move_mean=True, **
		featurizer = deepchem.feat.RawFeaturizer()
		elif featurizer == 'smiles2img':
		img_spec = kwargs.get("img_spec", "std")
		logger.info(img_spec)
		featurizer = deepchem.feat.SmilesToImage(img_spec=img_spec)

		loader = deepchem.data.CSVLoader(

+2 −3

Original line number	Diff line number	Diff line
		@@ -47,7 +47,6 @@ def load_tox21(featurizer='ECFP', split='index', reload=True, K=4, **kwargs):
		max_n_atoms=150, max_valence=6)
		elif featurizer == "smiles2img":
		img_spec = kwargs.get("img_spec", "std")
		logger.info(img_spec)
		featurizer = deepchem.feat.SmilesToImage(img_spec=img_spec)

		loader = deepchem.data.CSVLoader(