:ok_hand: add underscore for internal classes (71ccdb3a) · Commits · 钟慕尧 / deepchem

deepchem/data/datasets.py

+6 −6

Original line number	Diff line number	Diff line
		@@ -873,11 +873,11 @@ class NumpyDataset(Dataset):
		this dataset.
		"""
		try:
		from deepchem.data.pytorch_datasets import TorchNumpyDataset
		from deepchem.data.pytorch_datasets import _TorchNumpyDataset
		except:
		raise ValueError("This method requires PyTorch to be installed.")

		pytorch_ds = TorchNumpyDataset(
		pytorch_ds = _TorchNumpyDataset(
		numpy_dataset=self, epochs=epochs, deterministic=deterministic)
		return pytorch_ds

		@@ -1472,11 +1472,11 @@ class DiskDataset(Dataset):
		this dataset.
		"""
		try:
		from deepchem.data.pytorch_datasets import TorchDiskDataset
		from deepchem.data.pytorch_datasets import _TorchDiskDataset
		except:
		raise ValueError("This method requires PyTorch to be installed.")

		pytorch_ds = TorchDiskDataset(
		pytorch_ds = _TorchDiskDataset(
		disk_dataset=self, epochs=epochs, deterministic=deterministic)
		return pytorch_ds

		@@ -2246,11 +2246,11 @@ class ImageDataset(Dataset):
		this dataset.
		"""
		try:
		from deepchem.data.pytorch_datasets import TorchImageDataset
		from deepchem.data.pytorch_datasets import _TorchImageDataset
		except:
		raise ValueError("This method requires PyTorch to be installed.")

		pytorch_ds = TorchImageDataset(
		pytorch_ds = _TorchImageDataset(
		image_dataset=self, epochs=epochs, deterministic=deterministic)
		return pytorch_ds

+4 −4

Original line number	Diff line number	Diff line
		@@ -6,7 +6,7 @@ from deepchem.data.data_loader import ImageLoader
		from deepchem.data.datasets import NumpyDataset, DiskDataset, ImageDataset


		class TorchNumpyDataset(torch.utils.data.IterableDataset): # type: ignore
		class _TorchNumpyDataset(torch.utils.data.IterableDataset): # type: ignore

		def __init__(self, numpy_dataset: NumpyDataset, epochs: int,
		deterministic: bool):
		@@ -44,7 +44,7 @@ class TorchNumpyDataset(torch.utils.data.IterableDataset): # type: ignore
		self.numpy_dataset._w[i], self.numpy_dataset._ids[i])


		class TorchDiskDataset(torch.utils.data.IterableDataset): # type: ignore
		class _TorchDiskDataset(torch.utils.data.IterableDataset): # type: ignore

		def __init__(self, disk_dataset: DiskDataset, epochs: int,
		deterministic: bool):
		@@ -83,7 +83,7 @@ class TorchDiskDataset(torch.utils.data.IterableDataset): # type: ignore
		yield (X[i], y[i], w[i], ids[i])


		class TorchImageDataset(torch.utils.data.IterableDataset): # type: ignore
		class _TorchImageDataset(torch.utils.data.IterableDataset): # type: ignore

		def __init__(self, image_dataset: ImageDataset, epochs: int,
		deterministic: bool):
		@@ -123,7 +123,7 @@ class TorchImageDataset(torch.utils.data.IterableDataset): # type: ignore

		def _get_image(self, array: Union[np.ndarray, List[str]],
		index: int) -> np.ndarray:
		"""Function for loading an image
		"""Method for loading an image

		Parameters
		----------