[Typing] 修复示例中使用 Dataset 的 type-arg 类型检查错误 (#65895)

megemini · web-flow · commit 038c03443625 · 2024-07-13T10:39:14.000+08:00
diff --git a/python/paddle/distributed/auto_parallel/api.py b/python/paddle/distributed/auto_parallel/api.py
@@ -2378,7 +2378,7 @@ def to_static(
             >>> BATCH_NUM = 4
             >>> IMAGE_SIZE = 16
             >>> CLASS_NUM = 8
-            >>> class RandomDataset(paddle.io.Dataset):
+            >>> class RandomDataset(paddle.io.Dataset): # type: ignore[type-arg]
             ...     def __init__(self, images, labels, num_samples):
             ...         self.images = images
             ...         self.labels = labels
@@ -2885,7 +2885,7 @@ def shard_dataloader(
 
             >>> paddle.seed(1024)
             >>> np.random.seed(1024)
-            >>> class RandomDataset(Dataset):
+            >>> class RandomDataset(Dataset): # type: ignore[type-arg]
             >>>     def __init__(self, seq_len, hidden, num_samples=8):
             ...         super().__init__()
             ...         self.seq_len = seq_len
@@ -2904,10 +2904,10 @@ def shard_dataloader(
             ...     def __init__(self):
             ...         super(MlpModel, self).__init__()
             ...         self.w0 = dist.shard_tensor(
-            ...             self.create_parameter(shape=[HIDDLE_SIZE, HIDDLE_SIZE]),
+            ...             self.create_parameter(shape=[8, 8]),
             ...             mesh0, [dist.Replicate(), dist.Shard(1)])
             ...         self.w1 = dist.shard_tensor(
-            ...             self.create_parameter(shape=[HIDDLE_SIZE, HIDDLE_SIZE]),
+            ...             self.create_parameter(shape=[8, 8]),
             ...             mesh1, [dist.Replicate(), dist.Shard(0)])
 
             ...     def forward(self, x):
@@ -2978,7 +2978,7 @@ def shard_dataloader(
             >>> import numpy as np
             >>> mesh0 = dist.ProcessMesh([[0, 1], [2, 3]], dim_names=['dp', 'mp'])
             >>> mesh1 = dist.ProcessMesh([[4, 5], [6, 7]], dim_names=['dp', 'mp'])
-            >>> class RandomDataset(Dataset):
+            >>> class RandomDataset(Dataset): # type: ignore[type-arg]
             ...     def __init__(self, seq_len, hidden, num_samples=8):
             ...         super().__init__()
             ...         self.seq_len = seq_len
diff --git a/python/paddle/incubate/optimizer/lookahead.py b/python/paddle/incubate/optimizer/lookahead.py
@@ -63,7 +63,7 @@ class LookAhead(Optimizer):
             >>> IMAGE_SIZE = 784
             >>> 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
             ...     def __getitem__(self, idx):
diff --git a/python/paddle/incubate/optimizer/modelaverage.py b/python/paddle/incubate/optimizer/modelaverage.py
@@ -88,7 +88,7 @@ class ModelAverage(Optimizer):
             >>> 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
             ...     def __getitem__(self, idx):
diff --git a/python/paddle/io/dataloader/batch_sampler.py b/python/paddle/io/dataloader/batch_sampler.py
@@ -63,7 +63,7 @@ class BatchSampler(Sampler):
 
             >>> np.random.seed(2023)
             >>> # init with dataset
-            >>> class RandomDataset(Dataset):
+            >>> class RandomDataset(Dataset): # type: ignore[type-arg]
             ...     def __init__(self, num_samples):
             ...         self.num_samples = num_samples
             ...
diff --git a/python/paddle/io/dataloader/sampler.py b/python/paddle/io/dataloader/sampler.py
@@ -115,7 +115,7 @@ class SequenceSampler(Sampler):
             >>> import numpy as np
             >>> from paddle.io import Dataset, SequenceSampler
 
-            >>> class RandomDataset(Dataset):
+            >>> class RandomDataset(Dataset): # type: ignore[type-arg]
             ...     def __init__(self, num_samples):
             ...         self.num_samples = num_samples
             ...
diff --git a/python/paddle/jit/api.py b/python/paddle/jit/api.py
@@ -1001,7 +1001,7 @@ def save(
             >>> 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
             ...
@@ -1559,7 +1559,7 @@ def load(
                 >>> 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
                 ...
@@ -1652,7 +1652,7 @@ def load(
                 >>> 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/pir_translated_layer.py b/python/paddle/jit/pir_translated_layer.py
@@ -475,7 +475,7 @@ class PirTranslatedLayer(layers.Layer):
             >>> 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
@@ -1308,7 +1308,7 @@ class TranslatedLayer(layers.Layer):
             >>> 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/metric/metrics.py b/python/paddle/metric/metrics.py
@@ -387,7 +387,7 @@ class Precision(Metric):
             >>> import paddle
             >>> import paddle.nn as nn
 
-            >>> class Data(paddle.io.Dataset):
+            >>> class Data(paddle.io.Dataset): # type: ignore[type-arg]
             ...     def __init__(self):
             ...         super().__init__()
             ...         self.n = 1024
@@ -528,7 +528,7 @@ class Recall(Metric):
             >>> import paddle
             >>> import paddle.nn as nn
 
-            >>> class Data(paddle.io.Dataset):
+            >>> class Data(paddle.io.Dataset): # type: ignore[type-arg]
             ...     def __init__(self):
             ...         super().__init__()
             ...         self.n = 1024
@@ -677,7 +677,7 @@ class Auc(Metric):
             >>> import paddle
             >>> import paddle.nn as nn
 
-            >>> class Data(paddle.io.Dataset):
+            >>> class Data(paddle.io.Dataset): # type: ignore[type-arg]
             ...     def __init__(self):
             ...         super().__init__()
             ...         self.n = 1024
