File size: 293 Bytes
6ae852e
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
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]]