Skip to content

DataloaderBackends

jax_dataloader.core.DataloaderBackends ¤

Source code in jax_dataloader/core.py 
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
@dataclass(frozen=True)
class DataloaderBackends:
    jax = DataLoaderJAX
    pytorch: BaseDataLoader = DataLoaderPytorch
    tensorflow: BaseDataLoader = DataLoaderTensorflow
    merlin: BaseDataLoader = None

    __all__ = dict(
        jax=jax, pytorch=pytorch, tensorflow=tensorflow, merlin=merlin
    )

    def __getitem__(self, key):
        return self.__all__[key]

    @property
    def supported(self) -> List[str]:
        return [
            backend for backend, dl_cls in self.__all__.items() if dl_cls is not None
        ]

Attributes¤

__all__ = dict(jax=jax, pytorch=pytorch, tensorflow=tensorflow, merlin=merlin) class-attribute instance-attribute ¤

merlin: BaseDataLoader = None class-attribute instance-attribute ¤

supported: List[str] property ¤

Functions¤

__getitem__(key) ¤

Source code in jax_dataloader/core.py 
35
36
def __getitem__(self, key):
    return self.__all__[key]

options:
  filters: []


options:
  filters: []