[Typing][A-68] Add type annotations for paddle/regularizer.py (#65226)

DrRyanHuang · web-flow · commit 09b418e86bec · 2024-06-17T21:10:57.000+08:00
diff --git a/python/paddle/regularizer.py b/python/paddle/regularizer.py
@@ -13,6 +13,9 @@
 # limitations under the License.
 
 
+from __future__ import annotations
+
+import paddle
 from paddle import _C_ops, pir
 from paddle.base import framework
 from paddle.base.framework import in_dynamic_or_pir_mode
@@ -31,10 +34,12 @@ class WeightDecayRegularizer:
     of its implementations
     """
 
-    def __init__(self):
+    def __init__(self) -> None:
         pass
 
-    def __call__(self, param, grad, block):
+    def __call__(
+        self, param: paddle.Tensor, grad: paddle.Tensor, block: pir.Block
+    ):
         """Add corresponding weight decay operations to the network"""
         raise NotImplementedError()
 
@@ -105,12 +110,17 @@ class L1Decay(WeightDecayRegularizer):
             ...         bias_attr=False)
     """
 
-    def __init__(self, coeff=0.0):
+    def __init__(self, coeff: float = 0.0) -> None:
         assert coeff is not None
         super().__init__()
         self._coeff = coeff
 
-    def __call__(self, param, grad, block):
+    def __call__(
+        self,
+        param: paddle.Tensor,
+        grad: paddle.Tensor,
+        block: pir.Block,
+    ):
         """Add L1 weight decay ops to network
 
         Adds L1 weight decay ops.
@@ -152,7 +162,7 @@ def __call__(self, param, grad, block):
             )
             return decay
 
-    def __str__(self):
+    def __str__(self) -> str:
         return "L1Decay, coeff=%f" % self._coeff
 
 
@@ -217,12 +227,17 @@ class L2Decay(WeightDecayRegularizer):
             ...         bias_attr=False)
     """
 
-    def __init__(self, coeff=0.0):
+    def __init__(self, coeff: float = 0.0) -> None:
         assert coeff is not None
         super().__init__()
         self._coeff = coeff
 
-    def __call__(self, param, grad, block):
+    def __call__(
+        self,
+        param: paddle.Tensor,
+        grad: paddle.Tensor,
+        block: pir.Block,
+    ):
         """Add L2 weight decay ops to network
 
         Adds L2 weight decay ops.
@@ -257,5 +272,5 @@ def __call__(self, param, grad, block):
 
             return decay
 
-    def __str__(self):
+    def __str__(self) -> str:
         return "L2Decay, coeff=%f" % self._coeff
