Unverified Commit 361a1aab authored by Daiki Nishikawa's avatar Daiki Nishikawa Committed by GitHub
Browse files

Merge pull request #2103 from nd-02110114/update-dataloader-docs 

Update docstrings in data loader classes
parents ab911353 6604d60b

deepchem/data/__init__.py

+2 −0
Original line number Diff line number Diff line 
"""

Gathers all datasets in one place for convenient imports

"""

# flake8: noqa



# TODO(rbharath): Get rid of * import

from deepchem.data.datasets import pad_features

from deepchem.data.datasets import pad_batch

deepchem/data/data_loader.py

+187 −192

File changed.

Preview size limit exceeded, changes collapsed.

deepchem/data/datasets.py

+12 −15

File changed.

Preview size limit exceeded, changes collapsed.

deepchem/data/pytorch_datasets.py

+2 −2
Original line number Diff line number Diff line
@@ -2,7 +2,7 @@ from typing import List, Union 
import numpy as np

import torch



from deepchem.data.data_loader import ImageLoader

from deepchem.utils.save import load_image_files

from deepchem.data.datasets import NumpyDataset, DiskDataset, ImageDataset
@@ -139,4 +139,4 @@ class _TorchImageDataset(torch.utils.data.IterableDataset): # type: ignore 
    """

    if isinstance(array, np.ndarray):

      return array[index]

    return ImageLoader.load_img([array[index]])[0]
 
    return load_image_files([array[index]])[0]

deepchem/data/tests/test_csv_loader.py

+0 −2
Original line number Diff line number Diff line 
import os

from io import StringIO

import tempfile

import shutil

import deepchem as dc

CRA Git | Maintained and supported by SUSTech CRA and CCSE