嘉渊
2023-05-10 ad55e73198f51786c8f2f4a7dda5b34fd790de3f
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,