funasr/build_utils/build_streaming_iterator.py
@@ -18,14 +18,17 @@ dtype: str = np.float32, num_workers: int = 1, use_collate_fn: bool = True, preprocess_fn=None, ngpu: int = 0, train: bool=False, train: bool = False, ) -> DataLoader: """Build DataLoader using iterable dataset""" assert check_argument_types() # preprocess if preprocess_args is not None: if preprocess_fn is not None: preprocess_fn = preprocess_fn elif preprocess_args is not None: preprocess_args.task_name = task_name preprocess_fn = build_preprocess(preprocess_args, train) else:
