formt fixed.

Author: zhuyue

python/infinicore/nn/modules/__init__.py

@@ -1,3 +1,5 @@
 from .container import ModuleList
 from .module import Module
 from .parameter import Parameter
+
+__all__ = ["ModuleList", "Module", "Parameter"]

python/infinicore/nn/modules/module.py

@@ -575,7 +575,7 @@ def load(module, local_state_dict, prefix=""):
                         for k, v in local_state_dict.items()
                         if k.startswith(child_prefix)
                     }
-                    load(child, child_state_dict, child_prefix)
+                    load(child, child_state_dict, child_prefix)  # noqa: F821
 
         load(self, state_dict)
         del load