:rotating_light: fix lint erro (ef46193c) · Commits · 钟慕尧 / deepchem

deepchem/data/datasets.py

+2 −2

Original line number	Diff line number	Diff line
		@@ -339,11 +339,11 @@ class Dataset(object):

		def __repr__(self) -> str:
		"""Convert self to REPL print representation."""
		threshold = dc.utils.get_print_threshold()
		threshold = 10
		task_str = np.array2string(
		np.array(self.get_task_names()), threshold=threshold)
		X_shape, y_shape, w_shape, _ = self.get_shape()
		if self.__len__() < dc.utils.get_max_print_size():
		if self.__len__() < 1000:
		id_str = np.array2string(self.ids, threshold=threshold)
		return "<%s X.shape: %s, y.shape: %s, w.shape: %s, ids: %s, task_names: %s>" % (
		self.__class__.__name__, str(X_shape), str(y_shape), str(w_shape),

+4 −4

Original line number	Diff line number	Diff line
		@@ -136,8 +136,8 @@ def load_bace_regression(featurizer='ECFP',
		test = transformer.transform(test)

		if reload:
		deepchem.utils.data_utils.save_dataset_to_disk(save_folder, train, valid, test,
		transformers)
		deepchem.utils.data_utils.save_dataset_to_disk(save_folder, train, valid,
		test, transformers)
		return bace_tasks, (train, valid, test), transformers


		@@ -236,6 +236,6 @@ def load_bace_classification(featurizer='ECFP',
		test = transformer.transform(test)

		if reload:
		deepchem.utils.data_utils.save_dataset_to_disk(save_folder, train, valid, test,
		transformers)
		deepchem.utils.data_utils.save_dataset_to_disk(save_folder, train, valid,
		test, transformers)
		return bace_tasks, (train, valid, test), transformers

+4 −4

Original line number	Diff line number	Diff line
		@@ -94,8 +94,8 @@ def load_bbbc001(split='index',
		transformers = []
		all_dataset = (train, valid, test)
		if reload:
		deepchem.utils.data_utils.save_dataset_to_disk(save_folder, train, valid, test,
		transformers)
		deepchem.utils.data_utils.save_dataset_to_disk(save_folder, train, valid,
		test, transformers)
		return bbbc001_tasks, all_dataset, transformers


		@@ -177,6 +177,6 @@ def load_bbbc002(split='index',
		all_dataset = (train, valid, test)
		transformers = []
		if reload:
		deepchem.utils.data_utils.save_dataset_to_disk(save_folder, train, valid, test,
		transformers)
		deepchem.utils.data_utils.save_dataset_to_disk(save_folder, train, valid,
		test, transformers)
		return bbbc002_tasks, all_dataset, transformers

+2 −2

Original line number	Diff line number	Diff line
		@@ -123,6 +123,6 @@ def load_bbbp(featurizer='ECFP',
		test = transformer.transform(test)

		if reload:
		deepchem.utils.data_utils.save_dataset_to_disk(save_folder, train, valid, test,
		transformers)
		deepchem.utils.data_utils.save_dataset_to_disk(save_folder, train, valid,
		test, transformers)
		return bbbp_tasks, (train, valid, test), transformers

+2 −2

Original line number	Diff line number	Diff line
		@@ -72,6 +72,6 @@ def load_cell_counting(split=None,
		transformers = []
		all_dataset = (train, valid, test)
		if reload:
		deepchem.utils.data_utils.save_dataset_to_disk(save_folder, train, valid, test,
		transformers)
		deepchem.utils.data_utils.save_dataset_to_disk(save_folder, train, valid,
		test, transformers)
		return cell_counting_tasks, all_dataset, transformers