update comment

open-mmlab · Apr 27, 2023 · 60ebd01 · 60ebd01
1 parent 196616f
commit 60ebd01
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mmengine/registry/registry.py b/mmengine/registry/registry.py
@@ -591,6 +591,10 @@ def _register_module(self,
                 existed_module = self.module_dict[name]
                 raise KeyError(f'{name} is already registered in {self.name} '
                                f'at {existed_module.__module__}')
+            # If module is a partial function or user defined callable
+            # object, it may not have `__module__` and `__name__` attributes.
+            # We set the `__module__` and `__name__` attributes for them to
+            # provide more information.
             if not hasattr(module, '__module__'):
                 module_name = str(inspect.getmodule(sys._getframe(2)))
                 if module_name is None: