from typing import Dict, Sequence, TypeVar, Union

from deepscreen.data.utils.collator import collate_fn
from deepscreen.data.utils.label import label_transform
from deepscreen.data.utils.sampler import SafeBatchSampler

T = TypeVar('T')
FlexibleIterable = Union[T, Sequence[T], Dict[str, T]]