Skip to content
Snippets Groups Projects
Unverified Commit 38b22d9e authored by jbwang1997's avatar jbwang1997 Committed by GitHub
Browse files

[Enhance] Enhance error report when a module has been registered in registery. (#264)

* Update

* Add unittest
parent 172b9ded
No related branches found
No related tags found
No related merge requests found
......@@ -480,8 +480,9 @@ class Registry:
module_name = [module_name]
for name in module_name:
if not force and name in self._module_dict:
raise KeyError(f'{name} is already registered '
f'in {self.name}')
existed_module = self.module_dict[name]
raise KeyError(f'{name} is already registered in {self.name} '
f'at {existed_module.__module__}')
self._module_dict[name] = module_class
def register_module(
......
......@@ -115,7 +115,8 @@ class TestRegistry:
# force=False
with pytest.raises(
KeyError,
match='BritishShorthair is already registered in cat'):
match='BritishShorthair is already registered in cat '
'at test_registry'):
@CATS.register_module()
class BritishShorthair:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment