🐛Followed mypy

carefree0910 · Jun 16, 2022 · 8ca6408 · 8ca6408
1 parent f36c44e
commit 8ca6408
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cflearn/protocol.py b/cflearn/protocol.py
@@ -583,7 +583,7 @@ def register_(
         if tag in cls.d:
             return tag
 
-        @cls.register(tag)
+        @cls.register(tag)  # type: ignore
         class _(cls):  # type: ignore
             names = base_loss_names
 
@@ -653,7 +653,7 @@ def register_(
         if tag in cls.d:
             return tag
 
-        @cls.register(tag)
+        @cls.register(tag)  # type: ignore
         class _(cls):  # type: ignore
             loss_name = base_loss_name
             aux_names = aux_loss_names