<gh_stars>0
from dataclasses import dataclass
from typing import Optional

from torch.utils.data import DataLoader


@dataclass
class DataLoaderCollection:
    train_loader: DataLoader
    test_loader: DataLoader
    valid_loader: Optional[DataLoader] = None

from dataclasses import dataclass
from typing import Optional

from torch.utils.data import DataLoader


@dataclass
class DataLoaderCollection:
    train_loader: DataLoader
    test_loader: DataLoader
    valid_loader: Optional[DataLoader] = None