yapf (51480382) · Commits · 钟慕尧 / deepchem

deepchem/molnet/load_function/uspto_datasets.py

+5 −3

Original line number	Diff line number	Diff line
		@@ -11,7 +11,6 @@ from deepchem.molnet.load_function.molnet_loader import TransformerGenerator, _M
		from typing import List, Optional, Tuple, Union
		import deepchem as dc


		logger = logging.getLogger(__name__)

		DEFAULT_DIR = deepchem.utils.data_utils.get_data_dir()
		@@ -59,13 +58,16 @@ class _USPTOLoader(_MolnetLoader):
		dc.utils.data_utils.download_url(url=dataset_url, dest_dir=self.data_dir)
		logger.info("Dataset download complete.")

		loader = dc.data.CSVLoader(tasks=self.tasks, feature_field=["src", "tgt"], featurizer=self.featurizer)
		loader = dc.data.CSVLoader(
		tasks=self.tasks,
		feature_field=["src", "tgt"],
		featurizer=self.featurizer)

		return loader.create_dataset(dataset_file, shard_size=8192)


		def load_uspto(
		featurizer: Union[dc.feat.Featurizer, str] = dc.feat.DummyFeaturizer(), # This will be changed to dummy featurizer!
		featurizer: Union[dc.feat.Featurizer, str] = dc.feat.DummyFeaturizer(),
		splitter: Union[dc.splits.Splitter, str, None] = 'SpecifiedSplitter',
		transformers: List[Union[TransformerGenerator, str]] = [],
		reload: bool = True,