torchwrench.utils.data.dataset.wrapper module¶

class torchwrench.utils.data.dataset.wrapper.EmptyDataset[source]¶

Bases: Dataset[None]

Dataset placeholder. Raises StopIteration if __getitem__ is called.

class torchwrench.utils.data.dataset.wrapper.IterableSubset(dataset: SupportsGetitemLen[T, Any] | SupportsIterLen[T], indices: Iterable[int] | LongTensor1D)[source]¶

Bases: IterableWrapper[T], Generic[T]

class torchwrench.utils.data.dataset.wrapper.IterableTransformWrapper(dataset: SupportsGetitemLen[T, Any] | SupportsIterLen[T], transform: Callable[[T], U] | None, condition: Callable[[T, int], bool] | None = None)[source]¶

Bases: IterableWrapper[T], Generic[T, U]

property condition : Callable[[T, int], bool] | None¶

property transform : Callable[[T], U] | None¶

class torchwrench.utils.data.dataset.wrapper.IterableWrapper(dataset: SupportsGetitemLen[T, Any] | SupportsIterLen[T])[source]¶

Bases: Generic[T], IterableDataset[T], _WrapperBase[T]

class torchwrench.utils.data.dataset.wrapper.Subset(dataset: SupportsGetitemLen[T, Any], indices: Iterable[int])[source]¶

Bases: Generic[T], Subset[T], Wrapper[T]

class torchwrench.utils.data.dataset.wrapper.TransformWrapper(dataset: SupportsGetitemLen[T, Any], transform: Callable[[T], U] | None, condition: Callable[[T, int], bool] | None = None)[source]¶

Bases: Generic[T, U], Wrapper[T]

property condition : Callable[[T, int], bool] | None¶

property transform : Callable[[T], U] | None¶

class torchwrench.utils.data.dataset.wrapper.Wrapper(dataset: Any)[source]¶

Bases: Generic[T], _WrapperBase[T]