From b274af152df357e95f74889dc80a3fc866c727c1 Mon Sep 17 00:00:00 2001 From: yaoxuefeng6 Date: Thu, 17 Sep 2020 12:19:51 +0800 Subject: [PATCH] update new datagenerator in ut --- python/paddle/fluid/incubate/fleet/tests/ctr_dataset_reader.py | 2 +- python/paddle/fluid/tests/unittests/ctr_dataset_reader.py | 3 +-- python/paddle/fluid/tests/unittests/simnet_dataset_reader.py | 2 +- 3 files changed, 3 insertions(+), 4 deletions(-) diff --git a/python/paddle/fluid/incubate/fleet/tests/ctr_dataset_reader.py b/python/paddle/fluid/incubate/fleet/tests/ctr_dataset_reader.py index 32ba0e512f5..1407e92fd38 100644 --- a/python/paddle/fluid/incubate/fleet/tests/ctr_dataset_reader.py +++ b/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( diff --git a/python/paddle/fluid/tests/unittests/ctr_dataset_reader.py b/python/paddle/fluid/tests/unittests/ctr_dataset_reader.py index 15e98481c26..6799d943df1 100644 --- a/python/paddle/fluid/tests/unittests/ctr_dataset_reader.py +++ b/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) diff --git a/python/paddle/fluid/tests/unittests/simnet_dataset_reader.py b/python/paddle/fluid/tests/unittests/simnet_dataset_reader.py index 41eadc13a2a..dea23ae97f3 100644 --- a/python/paddle/fluid/tests/unittests/simnet_dataset_reader.py +++ b/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") -- GitLab