update new datagenerator in ut

python/paddle/fluid/incubate/fleet/tests/ctr_dataset_reader.py

@@ -19,7 +19,7 @@ import tarfile
 import os
 
 import paddle
-import paddle.fluid.incubate.data_generator as data_generator
+from paddle.distributed.fleet.dataset import data_generator as data_generator
 from paddle.fluid.log_helper import get_logger
 
 logger = get_logger(

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

@@ -22,8 +22,7 @@ import random
 import warnings
 
 import paddle
-import paddle.fluid.incubate.data_generator as data_generator
-
+from paddle.distributed.fleet.dataset import data_generator as data_generator
 logging.basicConfig()
 logger = logging.getLogger("paddle")
 logger.setLevel(logging.INFO)

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

@@ -21,7 +21,7 @@ import tarfile
 import random
 
 import paddle
-import paddle.fluid.incubate.data_generator as data_generator
+from paddle.distributed.fleet.dataset import data_generator as data_generator
 
 logging.basicConfig()
 logger = logging.getLogger("paddle")