[Typing] Fix undefined names in example code (#65429)

SigureMo · web-flow · commit 534decdf2c42 · 2024-06-25T10:09:13.000+08:00
diff --git a/python/paddle/base/framework.py b/python/paddle/base/framework.py
@@ -1798,6 +1798,7 @@ def numpy(self):
         Examples:
             .. code-block:: python
 
+                >>> import paddle
                 >>> import paddle.base as base
                 >>> from paddle.nn import Linear
                 >>> import numpy as np
diff --git a/python/paddle/distributed/auto_parallel/api.py b/python/paddle/distributed/auto_parallel/api.py
@@ -2251,9 +2251,11 @@ def state_dict(self, mode="all"):
         ).state_dict(mode)
         dist_state_dict = self._build_distributed_state_dict(local_state_dict)
         mapping_names = [
-            self._parameter_to_structured_name[k]
-            if k in self._parameter_to_structured_name
-            else k
+            (
+                self._parameter_to_structured_name[k]
+                if k in self._parameter_to_structured_name
+                else k
+            )
             for k in dist_state_dict.keys()
         ]
         dist_state_dict = dict(
@@ -2881,6 +2883,8 @@ def shard_dataloader(
         .. code-block:: python
             :name: example-1
 
+            >>> import os
+            >>> import numpy as np
             >>> import paddle
             >>> import paddle.distributed as dist
             >>> from paddle.io import BatchSampler, DataLoader, Dataset
diff --git a/python/paddle/distributed/communication/stream/gather.py b/python/paddle/distributed/communication/stream/gather.py
@@ -89,7 +89,7 @@ def gather(
             ...     dist.stream.gather(data, gather_list, dst=0)
             >>> else:
             ...     data = paddle.to_tensor([4, 5, 6])
-            ...     dist.stream.gather(data1, gather_list, dst=0)
+            ...     dist.stream.gather(data, gather_list, dst=0)
             >>> print(gather_list)
             >>> # [[1, 2, 3], [4, 5, 6]] (2 GPUs, out for rank 0)
             >>> # [] (2 GPUs, out for rank 1)
diff --git a/python/paddle/io/dataloader/sampler.py b/python/paddle/io/dataloader/sampler.py
@@ -45,6 +45,7 @@ class Sampler:
 
         .. code-block:: python
 
+            >>> import numpy as np
             >>> from paddle.io import Dataset, Sampler
 
             >>> class RandomDataset(Dataset):
@@ -111,6 +112,7 @@ class SequenceSampler(Sampler):
 
         .. code-block:: python
 
+            >>> import numpy as np
             >>> from paddle.io import Dataset, SequenceSampler
 
             >>> class RandomDataset(Dataset):
diff --git a/python/paddle/metric/metrics.py b/python/paddle/metric/metrics.py
@@ -85,6 +85,8 @@ class Metric(metaclass=abc.ABCMeta):
         .. code-block:: python
             :name: code-compute-example
 
+            >>> import paddle
+
             >>> def compute(pred, label):
             ...     # sort prediction and slice the top-5 scores
             ...     pred = paddle.argsort(pred, descending=True)[:, :5]
diff --git a/python/paddle/nn/utils/spectral_norm_hook.py b/python/paddle/nn/utils/spectral_norm_hook.py
@@ -184,6 +184,7 @@ def spectral_norm(
     Examples:
         .. code-block:: python
 
+            >>> import paddle
             >>> from paddle.nn import Conv2D
             >>> from paddle.nn.utils import spectral_norm
             >>> paddle.seed(2023)
