From 580b11b57ac4b62f7e2acda73813a4e10e8e4cd3 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 10 十月 2023 17:17:29 +0800
Subject: [PATCH] v0.8.0
---
funasr/samplers/build_batch_sampler.py | 7 ++-----
1 files changed, 2 insertions(+), 5 deletions(-)
diff --git a/funasr/samplers/build_batch_sampler.py b/funasr/samplers/build_batch_sampler.py
index edda6ba..9266fea 100644
--- a/funasr/samplers/build_batch_sampler.py
+++ b/funasr/samplers/build_batch_sampler.py
@@ -1,10 +1,9 @@
from typing import List
+from typing import Dict
from typing import Sequence
from typing import Tuple
from typing import Union
-from typeguard import check_argument_types
-from typeguard import check_return_type
from funasr.samplers.abs_sampler import AbsSampler
from funasr.samplers.folded_batch_sampler import FoldedBatchSampler
@@ -78,7 +77,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,
@@ -103,7 +102,6 @@
padding: Whether sequences are input as a padded tensor or not.
used for "numel" mode
"""
- assert check_argument_types()
if len(shape_files) == 0:
raise ValueError("No shape file are given")
@@ -163,5 +161,4 @@
else:
raise ValueError(f"Not supported: {type}")
- assert check_return_type(retval)
return retval
--
Gitblit v1.9.1