zhaomingwork
2023-05-19 b63e73ae4f5df9d4ed9fb0bee12ac2cc09d7f523
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,