Fix black

2021-04-23 02:14:49 -07:00
parent 9b895bdf2e
commit 77c250c8fc
2 changed files with 13 additions and 9 deletions
--- a/lib/xlayers/super_module.py
+++ b/lib/xlayers/super_module.py
@@ -49,8 +49,8 @@ class SuperModule(abc.ABC, nn.Module):
    def add_module(self, name: str, module: Optional[torch.nn.Module]) -> None:
        if not isinstance(module, SuperModule):
            warnings.warn(
-                "Add {:} module, which is not SuperModule, into {:}".format(
-                    name, self.__class__.__name__
+                "Add {:}:{:} module, which is not SuperModule, into {:}".format(
+                    name, module.__class__.__name__, self.__class__.__name__
                )
                + "\n"
                + "It may cause some functions invalid."