speech_asr
2023-02-14 67c3cf83adcadc8e4d70dac65cee3ee4cbfe7748
funasr/samplers/build_batch_sampler.py
@@ -1,4 +1,5 @@
from typing import List
from typing import Dict
from typing import Sequence
from typing import Tuple
from typing import Union
@@ -78,7 +79,7 @@
    type: str,
    batch_size: int,
    batch_bins: int,
    shape_files: Union[Tuple[str, ...], List[str]],
    shape_files: Union[Tuple[str, ...], List[str], Dict],
    sort_in_batch: str = "descending",
    sort_batch: str = "ascending",
    drop_last: bool = False,