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

deepchem/data/data_loader.py

+8 −3

Original line number	Diff line number	Diff line
		@@ -372,14 +372,19 @@ class ImageLoader(DataLoader):
		filename, extension = os.path.splitext(input_file)
		# TODO(rbharath): Add support for more extensions
		if os.path.isdir(input_file):
		dirfiles = [os.path.join(input_file, subfile) for subfile in os.listdir(input_file)]
		dirfiles = [
		os.path.join(input_file, subfile)
		for subfile in os.listdir(input_file)
		]
		remainder += dirfiles
		elif extension == ".zip":
		zip_dir = tempfile.mkdtemp()
		zip_ref = zipfile.ZipFile(input_file, 'r')
		zip_ref.extractall(path=zip_dir)
		zip_ref.close()
		zip_files = [os.path.join(zip_dir, name) for name in zip_ref.namelist()]
		zip_files = [
		os.path.join(zip_dir, name) for name in zip_ref.namelist()
		]
		for zip_file in zip_files:
		_, extension = os.path.splitext(zip_file)
		if extension in [".png", ".tif"]:

+4 −5

Original line number	Diff line number	Diff line
		@@ -38,8 +38,7 @@ def load_bbbc001(split='index', reload=True):
		)
		if not os.path.exists(labels_file):
		deepchem.utils.download_url(
		'https://data.broadinstitute.org/bbbc/BBBC001/BBBC001_v1_counts.txt'
		)
		'https://data.broadinstitute.org/bbbc/BBBC001/BBBC001_v1_counts.txt')
		# Featurize Images into NumpyArrays
		loader = deepchem.data.ImageLoader()
		dataset = loader.featurize(dataset_file, in_memory=False)

+1 −1

File changed.

Contains only whitespace changes.