scvi-tools/src/scvi/dataloaders/__init__.py at 70a2e279850e40b2e895b0d2e6dd26920a1025e8 · scverse/scvi-tools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# for backwards compatibility, this was moved to scvi.data
from scvi.data import AnnTorchDataset

from ._ann_dataloader import AnnDataLoader
from ._anncollection import CollectionAdapter
from ._concat_dataloader import ConcatDataLoader
from ._custom_dataloaders import (
    AnnbatchDataModule,
    MappedCollectionDataModule,
    TileDBDataModule,
)
from ._data_splitting import (
    DataSplitter,
    DeviceBackedDataSplitter,
    SemiSupervisedDataSplitter,
)
from ._samplers import BatchDistributedSampler
from ._semi_dataloader import SemiSupervisedDataLoader

__all__ = [
    "AnnDataLoader",
    "AnnTorchDataset",
    "CollectionAdapter",
    "ConcatDataLoader",
    "DeviceBackedDataSplitter",
    "SemiSupervisedDataLoader",
    "DataSplitter",
    "SemiSupervisedDataSplitter",
    "BatchDistributedSampler",
    "AnnbatchDataModule",
    "MappedCollectionDataModule",
    "TileDBDataModule",
]