fix IterableDataset doc. test=document_fix (#32544)

PaddlePaddle · Apr 26, 2021 · fd85a4a · fd85a4a
1 parent 5161f71
commit fd85a4a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/paddle/fluid/dataloader/dataset.py b/python/paddle/fluid/dataloader/dataset.py
@@ -97,10 +97,10 @@ class IterableDataset(Dataset):
         .. code-block:: python
 
             import numpy as np
-            from paddle.io import Dataset
+            from paddle.io import IterableDataset
             
             # define a random dataset
-            class RandomDataset(Dataset):
+            class RandomDataset(IterableDataset):
                 def __init__(self, num_samples):
                     self.num_samples = num_samples