Skip to content

Commit 53a06e3

Browse files
committed
fix(register): 修改注册重复键值的处理方式
更新注册系统,在尝试注册已存在的键值时,从抛出异常改为打印提示信息并重新注册。
1 parent 9118496 commit 53a06e3

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

funasr/register.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,11 @@
33
from dataclasses import dataclass
44
import re
55

6+
67
@dataclass
78
class RegisterTables:
89
"""Registry system for classes."""
10+
911
model_classes = {}
1012
frontend_classes = {}
1113
specaug_classes = {}
@@ -46,6 +48,7 @@ def print(self, key: str = None) -> None:
4648

4749
def register(self, register_tables_key: str, key: str = None) -> callable:
4850
"""Decorator to register a class."""
51+
4952
def decorator(target_class):
5053
if not hasattr(self, register_tables_key):
5154
setattr(self, register_tables_key, {})
@@ -55,7 +58,7 @@ def decorator(target_class):
5558
registry_key = key if key is not None else target_class.__name__
5659

5760
if registry_key in registry:
58-
raise ValueError(f"Key {registry_key} already exists in {register_tables_key}")
61+
print(f"Key {registry_key} already exists in {register_tables_key}, re-register")
5962

6063
registry[registry_key] = target_class
6164

@@ -78,4 +81,5 @@ def decorator(target_class):
7881

7982
return decorator
8083

84+
8185
tables = RegisterTables()

funasr/version.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
1.1.13
1+
1.1.14

0 commit comments

Comments
 (0)