File size: 549 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from typing import Any, Iterator, TypeVar
T = TypeVar('T')
# These are only mock implementations when torch is not available
# In real usage, they will be imported from torch.utils.data
class IterableDataset:
def __iter__(self) -> Iterator[Any]: ...
class DataLoader:
def __init__(self, *args: Any, **kwargs: Any) -> None: ...
def __iter__(self) -> Iterator[Any]: ...
# Mock implementation of TorchTensor when torch is not available
class TorchTensor:
shape: tuple
def __init__(self, *args: Any, **kwargs: Any) -> None: ...
|