[Typing][B-86] Add type annotations for python/paddle/nn/utils/clip_grad_value_.py (#65809)

megemini · web-flow · commit 0786e23c0c64 · 2024-07-10T13:53:32.000+08:00
diff --git a/python/paddle/nn/utils/clip_grad_value_.py b/python/paddle/nn/utils/clip_grad_value_.py
@@ -12,16 +12,23 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Iterable
+
 import paddle
 
+if TYPE_CHECKING:
+    from paddle import Tensor
+
 __all__ = []
 
 
 @paddle.autograd.no_grad()
 def clip_grad_value_(
-    parameters,
-    clip_value,
-):
+    parameters: Iterable[Tensor] | Tensor,
+    clip_value: float,
+) -> None:
     r"""
     Clips gradient of an iterable of parameters at specified value.
     The gradient will be modified in place.
