[Typing][A-3] Add type annotations for paddle/tensor/creation.py

python/paddle/_typing/__init__.py

@@ -44,7 +44,6 @@
 
 # Shape
 from .shape import (
-    DynamicShapeLike as DynamicShapeLike,
     ShapeLike as ShapeLike,
     Size1 as Size1,
     Size2 as Size2,

python/paddle/_typing/shape.py

@@ -13,26 +13,26 @@
 # limitations under the License.
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, List, Tuple, Union
+from typing import TYPE_CHECKING, List, Sequence, Tuple, Union
 
 from typing_extensions import TypeAlias
 
 if TYPE_CHECKING:
     from .. import Tensor
 
-DynamicShapeLike: TypeAlias = Union[
-    Tuple[Union[int, "Tensor", None], ...],
-    List[Union[int, "Tensor", None]],
+
+_DynamicShapeLike: TypeAlias = Union[
+    Sequence[Union[int, "Tensor", None]],
     "Tensor",
 ]
 
 
-ShapeLike: TypeAlias = Union[
-    Tuple[int, ...],
-    List[int],
+_StaticShapeLike: TypeAlias = Union[
+    Sequence[int],
     "Tensor",
 ]
 
+ShapeLike: TypeAlias = Union[_DynamicShapeLike, _StaticShapeLike]
 
 # for size parameters, eg, kernel_size, stride ...
 Size1: TypeAlias = Union[int, Tuple[int], List[int]]

python/paddle/base/data_feeder.py

@@ -17,6 +17,7 @@
 import numpy as np
 
 from paddle import pir
+from paddle._typing.dtype_like import DTypeLike
 
 from ..pir import Value
 from ..pir.core import _PADDLE_PIR_DTYPE_2_NUMPY_DTYPE, ParameterMeta
@@ -89,7 +90,7 @@ def convert_uint16_to_float(data):
     return np.reshape(new_data, data.shape)
 
 
-def convert_dtype(dtype):
+def convert_dtype(dtype: DTypeLike) -> str:
     if isinstance(dtype, core.VarDesc.VarType):
         if dtype in _PADDLE_DTYPE_2_NUMPY_DTYPE:
             return _PADDLE_DTYPE_2_NUMPY_DTYPE[dtype]