[Typing] 修复 exe.run 以及示例中的类型标注错误 (#67302)

megemini · web-flow · commit ec2fcd008d53 · 2024-08-12T10:15:27.000+08:00
diff --git a/python/paddle/base/executor.py b/python/paddle/base/executor.py
@@ -54,7 +54,7 @@
 from .wrapped_decorator import signature_safe_contextmanager
 
 if TYPE_CHECKING:
-    from collections.abc import Generator
+    from collections.abc import Generator, Sequence
 
     import numpy.typing as npt
 
@@ -1675,7 +1675,7 @@ def run(
         self,
         program: Program | CompiledProgram | None = ...,
         feed: dict[str, npt.NDArray[Any]] | list[npt.NDArray[Any]] | None = ...,
-        fetch_list: list[str | Tensor] | None = ...,
+        fetch_list: str | Tensor | Sequence[str | Tensor] | None = ...,
         feed_var_name: str = ...,
         fetch_var_name: str = ...,
         scope: core.Scope | None = ...,
@@ -1689,7 +1689,7 @@ def run(
         self,
         program: Program | CompiledProgram | None = ...,
         feed: dict[str, npt.NDArray[Any]] | list[npt.NDArray[Any]] | None = ...,
-        fetch_list: list[str | Tensor] | None = ...,
+        fetch_list: str | Tensor | Sequence[str | Tensor] | None = ...,
         feed_var_name: str = ...,
         fetch_var_name: str = ...,
         scope: core.Scope | None = ...,
@@ -1703,7 +1703,7 @@ def run(
         self,
         program: Program | CompiledProgram | None = ...,
         feed: dict[str, npt.NDArray[Any]] | list[npt.NDArray[Any]] | None = ...,
-        fetch_list: list[str | Tensor] | None = ...,
+        fetch_list: str | Tensor | Sequence[str | Tensor] | None = ...,
         feed_var_name: str = ...,
         fetch_var_name: str = ...,
         scope: core.Scope | None = ...,
diff --git a/python/paddle/device/__init__.py b/python/paddle/device/__init__.py
@@ -975,7 +975,7 @@ class stream_guard:
             >>> data2 = paddle.ones(shape=[20])
             >>> data3 = data1 + data2
             >>> with paddle.device.stream_guard(s):
-            ...     s.wait_stream(paddle.device.default_stream())
+            ...     s.wait_stream(paddle.device.default_stream()) # type: ignore[attr-defined]
             ...     data4 = data1 + data3
 
     '''
diff --git a/python/paddle/distributed/communication/recv.py b/python/paddle/distributed/communication/recv.py
@@ -97,7 +97,7 @@ def irecv(
             >>> else:
             ...     data = paddle.to_tensor([1, 2, 3])
             ...     task = dist.irecv(data, src=0)
-            >>> task.wait()
+            >>> task.wait()  # type: ignore[union-attr]
             >>> print(data)
             >>> # [7, 8, 9] (2 GPUs)
     """
diff --git a/python/paddle/incubate/jit/inference_decorator.py b/python/paddle/incubate/jit/inference_decorator.py
@@ -564,7 +564,7 @@ def inference(
             >>> batch = 4096
             >>> hidd = 1024
             >>> dtype = "bfloat16"
-            >>> x = paddle.rand([batch, hidd], dtype=dtype)
+            >>> x = paddle.rand([batch, hidd], dtype=dtype) # type: ignore[arg-type]
             >>> mylayer = ExampleLayer(hidd)
             >>> dynamic_result = mylayer(x)
             >>> mylayer = paddle.incubate.jit.inference(mylayer)
diff --git a/python/paddle/jit/pir_translated_layer.py b/python/paddle/jit/pir_translated_layer.py
@@ -684,7 +684,7 @@ def program(self, method_name='forward'):
                 >>> CLASS_NUM = 10
 
                 >>> # define a random dataset
-                >>> class RandomDataset(paddle.io.Dataset):
+                >>> class RandomDataset(paddle.io.Dataset): # type: ignore[type-arg]
                 ...     def __init__(self, num_samples):
                 ...         self.num_samples = num_samples
                 ...
diff --git a/python/paddle/jit/translated_layer.py b/python/paddle/jit/translated_layer.py
@@ -1518,7 +1518,7 @@ def program(self, method_name='forward'):
                 >>> CLASS_NUM = 10
 
                 >>> # define a random dataset
-                >>> class RandomDataset(paddle.io.Dataset):
+                >>> class RandomDataset(paddle.io.Dataset): # type: ignore[type-arg]
                 ...     def __init__(self, num_samples):
                 ...         self.num_samples = num_samples
                 ...
diff --git a/python/paddle/profiler/profiler.py b/python/paddle/profiler/profiler.py
@@ -428,7 +428,7 @@ class Profiler:
                 >>> import paddle
                 >>> import paddle.profiler as profiler
 
-                >>> class RandomDataset(paddle.io.Dataset):
+                >>> class RandomDataset(paddle.io.Dataset): # type: ignore[type-arg]
                 ...     def __init__(self, num_samples):
                 ...         self.num_samples = num_samples
                 ...     def __getitem__(self, idx):
