[Typing] 修改一些示例代码中的类型错误 part 2 (#65463)

megemini · web-flow · commit b5087ad1dd15 · 2024-06-27T16:43:39.000+08:00
diff --git a/python/paddle/incubate/autograd/primx.py b/python/paddle/incubate/autograd/primx.py
@@ -649,7 +649,7 @@ def prim2orig(block=None, blacklist=None):
             >>> enable_prim()
 
             >>> x = paddle.ones(shape=[2, 2], dtype='float32')
-            >>> x.stop_gradients = False
+            >>> x.stop_gradient = False
             >>> y = x * x
             >>> dy_dx = paddle.static.gradients(y, x)
             >>> if prim_enabled():
diff --git a/python/paddle/nn/functional/common.py b/python/paddle/nn/functional/common.py
@@ -2271,7 +2271,7 @@ class centers and the shape of sampled_class_center will be [num_positive_class_
         >>> # num_classes of each GPU can be different, e.g num_classes_list = [10, 8]
         >>> num_classes_list = [10, 10]
         >>> num_classes = paddle.sum(paddle.to_tensor(num_classes_list))
-        >>> label = paddle.randint(low=0, high=num_classes.item(), shape=[batch_size], dtype='int64')
+        >>> label = paddle.randint(low=0, high=num_classes.item(), shape=[batch_size], dtype='int64') # type: ignore
         >>> label_list = [] # type: ignore
         >>> dist.all_gather(label_list, label)
         >>> label = paddle.concat(label_list, axis=0)
diff --git a/python/paddle/nn/layer/container.py b/python/paddle/nn/layer/container.py
@@ -605,7 +605,7 @@ def __init__(self, *layers: Layer | tuple[str, Layer] | list[Any]) -> None:
             for idx, layer in enumerate(layers):
                 self.add_sublayer(str(idx), layer)
 
-    def __getitem__(self, name: str) -> Layer:
+    def __getitem__(self, name: str | slice | int) -> Layer:
         if isinstance(name, slice):
             return self.__class__(*(list(self._sub_layers.values())[name]))
         elif isinstance(name, str):
