fix ut fail

python/paddle/distributed/fleet/dataset/data_generator.py

@@ -45,7 +45,7 @@ class DataGenerator(object):
         Example:
 
             .. code-block:: python
-                import paddle.fluid.incubate.data_generator as dg
+                from paddle.distributed.fleet.dataset import data_generator as dg
                 class MyData(dg.DataGenerator):
 
                     def generate_sample(self, line):
@@ -71,7 +71,7 @@ class DataGenerator(object):
 
         Example:
             .. code-block:: python
-                import paddle.fluid.incubate.data_generator as dg
+                from paddle.distributed.fleet.dataset import data_generator as dg
                 class MyData(dg.DataGenerator):
 
                     def generate_sample(self, line):
@@ -109,7 +109,7 @@ class DataGenerator(object):
         Example:
         
             .. code-block:: python
-                import paddle.fluid.incubate.data_generator as dg
+                from paddle.distributed.fleet.dataset import data_generator as dg
                 class MyData(dg.DataGenerator):
 
                     def generate_sample(self, line):
@@ -180,7 +180,7 @@ class DataGenerator(object):
         Example:
 
             .. code-block:: python
-                import paddle.fluid.incubate.data_generator as dg
+                from paddle.distributed.fleet.dataset import data_generator as dg
                 class MyData(dg.DataGenerator):
 
                     def generate_sample(self, line):
@@ -211,7 +211,7 @@ class DataGenerator(object):
         Example:
 
             .. code-block:: python
-                import paddle.fluid.incubate.data_generator as dg
+                from paddle.distributed.fleet.dataset import data_generator as dg
                 class MyData(dg.DataGenerator):
 
                     def generate_sample(self, line):

python/paddle/distributed/fleet/dataset/test_data_generator.py

@@ -10,10 +10,10 @@
 # distributed under the License is distributed on an "AS IS" BASIS,
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
-from paddle.distributed.fleet.dataset.data_generator import *
+from paddle.distributed.fleet.dataset import data_generator as dg
 
 
-class SyntheticData(MultiSlotDataGenerator):
+class SyntheticData(dg.MultiSlotDataGenerator):
     def generate_sample(self, line):
         def data_iter():
             for i in range(10000):
@@ -22,7 +22,7 @@ class SyntheticData(MultiSlotDataGenerator):
         return data_iter
 
 
-class SyntheticStringData(MultiSlotStringDataGenerator):
+class SyntheticStringData(dg.MultiSlotStringDataGenerator):
     def generate_sample(self, line):
         def data_iter():
             for i in range(10000):

python/paddle/fluid/tests/unittests/test_dataset.py

@@ -202,7 +202,7 @@ class TestDataset(unittest.TestCase):
         else:
             for i in range(self.epoch_num):
                 try:
-                    exe.train_from_dataset(fluid.main_program, dataset)
+                    exe.train_from_dataset(main_program, dataset)
                 except Exception as e:
                     self.assertTrue(False)