From 2e36e738ca39afc8d02f3d11013bd12f937cc874 Mon Sep 17 00:00:00 2001
From: zhaomingwork <61895407+zhaomingwork@users.noreply.github.com>
Date: 星期三, 08 十一月 2023 09:22:06 +0800
Subject: [PATCH] fix bug for h5 hotwords (#1067)
---
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