More doctests fixes (01c40bd3) · Commits · 钟慕尧 / deepchem

deepchem/data/datasets.py

+9 −9

Original line number	Diff line number	Diff line
		@@ -199,9 +199,9 @@ class Dataset(object):

		>>> dataset = NumpyDataset(np.ones((2,2)))
		>>> for x, y, w, id in dataset.itersamples():
		... print(x, y, w, id)
		[1. 1.] [0.] [0.] 0
		[1. 1.] [0.] [0.] 1
		... print(x.tolist(), y.tolist(), w.tolist(), id)
		[1.0 1.0] [0.0] [0.0] 0
		[1.0 1.0] [0.0] [0.0] 1
		"""
		raise NotImplementedError()

		@@ -408,9 +408,9 @@ class NumpyDataset(Dataset):

		>>> dataset = NumpyDataset(np.ones((2,2)))
		>>> for x, y, w, id in dataset.itersamples():
		... print(x, y, w, id)
		[1. 1.] [0.] [0.] 0
		[1. 1.] [0.] [0.] 1
		... print(x.tolist(), y.tolist(), w.tolist(), id)
		[1.0 1.0] [0.0] [0.0] 0
		[1.0 1.0] [0.0] [0.0] 1
		"""
		n_samples = self._X.shape[0]
		return ((self._X[i], self._y[i], self._w[i], self._ids[i])
		@@ -888,9 +888,9 @@ class DiskDataset(Dataset):

		>>> dataset = DiskDataset.from_numpy(np.ones((2,2)), np.ones((2,1)), verbose=False)
		>>> for x, y, w, id in dataset.itersamples():
		... print(x, y, w, id)
		[1. 1.] [1.] [1.] 0
		[1. 1.] [1.] [1.] 1
		... print(x.tolist(), y.tolist(), w.tolist(), id)
		[1.0 1.0] [0.0] [0.0] 0
		[1.0 1.0] [0.0] [0.0] 1
		"""

		def iterate(dataset):