From 98abc0e5ac1a1da0fe1802d9ffb623802fbf0b2f Mon Sep 17 00:00:00 2001
From: jmwang66 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期四, 29 六月 2023 16:30:39 +0800
Subject: [PATCH] update setup (#686)
---
funasr/train/reporter.py | 9 ---------
1 files changed, 0 insertions(+), 9 deletions(-)
diff --git a/funasr/train/reporter.py b/funasr/train/reporter.py
index 2921fef..cfe31f5 100644
--- a/funasr/train/reporter.py
+++ b/funasr/train/reporter.py
@@ -18,8 +18,6 @@
import humanfriendly
import numpy as np
import torch
-from typeguard import check_argument_types
-from typeguard import check_return_type
Num = Union[float, int, complex, torch.Tensor, np.ndarray]
@@ -27,7 +25,6 @@
def to_reported_value(v: Num, weight: Num = None) -> "ReportedValue":
- assert check_argument_types()
if isinstance(v, (torch.Tensor, np.ndarray)):
if np.prod(v.shape) != 1:
raise ValueError(f"v must be 0 or 1 dimension: {len(v.shape)}")
@@ -42,12 +39,10 @@
retval = WeightedAverage(v, weight)
else:
retval = Average(v)
- assert check_return_type(retval)
return retval
def aggregate(values: Sequence["ReportedValue"]) -> Num:
- assert check_argument_types()
for v in values:
if not isinstance(v, type(values[0])):
@@ -86,7 +81,6 @@
else:
raise NotImplementedError(f"type={type(values[0])}")
- assert check_return_type(retval)
return retval
@@ -122,7 +116,6 @@
"""
def __init__(self, key: str, epoch: int, total_count: int):
- assert check_argument_types()
self.key = key
self.epoch = epoch
self.start_time = time.perf_counter()
@@ -160,7 +153,6 @@
stats: Dict[str, Optional[Union[Num, Dict[str, Num]]]],
weight: Num = None,
) -> None:
- assert check_argument_types()
if self._finished:
raise RuntimeError("Already finished")
if len(self._seen_keys_in_the_step) == 0:
@@ -293,7 +285,6 @@
"""
def __init__(self, epoch: int = 0):
- assert check_argument_types()
if epoch < 0:
raise ValueError(f"epoch must be 0 or more: {epoch}")
self.epoch = epoch
--
Gitblit v1.9.1