File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change 3
3
from dataclasses import dataclass
4
4
import re
5
5
6
+
6
7
@dataclass
7
8
class RegisterTables :
8
9
"""Registry system for classes."""
10
+
9
11
model_classes = {}
10
12
frontend_classes = {}
11
13
specaug_classes = {}
@@ -46,6 +48,7 @@ def print(self, key: str = None) -> None:
46
48
47
49
def register (self , register_tables_key : str , key : str = None ) -> callable :
48
50
"""Decorator to register a class."""
51
+
49
52
def decorator (target_class ):
50
53
if not hasattr (self , register_tables_key ):
51
54
setattr (self , register_tables_key , {})
@@ -55,7 +58,7 @@ def decorator(target_class):
55
58
registry_key = key if key is not None else target_class .__name__
56
59
57
60
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 " )
59
62
60
63
registry [registry_key ] = target_class
61
64
@@ -78,4 +81,5 @@ def decorator(target_class):
78
81
79
82
return decorator
80
83
84
+
81
85
tables = RegisterTables ()
Original file line number Diff line number Diff line change 1
- 1.1.13
1
+ 1.1.14
You can’t perform that action at this time.
0 commit comments