嘉渊
2023-05-12 bc6fd77c94e43275807ed0355f34350219d1cdd0
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,