From 1e1a7e5c828653c050480fc1e20ab9957a2e1c76 Mon Sep 17 00:00:00 2001 From: yaoxuefeng6 Date: Thu, 17 Sep 2020 11:32:09 +0800 Subject: [PATCH] fix ut fail --- .../paddle/distributed/fleet/dataset/data_generator.py | 10 +++++----- .../distributed/fleet/dataset/test_data_generator.py | 6 +++--- python/paddle/fluid/tests/unittests/test_dataset.py | 2 +- 3 files changed, 9 insertions(+), 9 deletions(-) diff --git a/python/paddle/distributed/fleet/dataset/data_generator.py b/python/paddle/distributed/fleet/dataset/data_generator.py index 5d0018ca4e7..77c60c8dd30 100644 --- a/python/paddle/distributed/fleet/dataset/data_generator.py +++ b/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): diff --git a/python/paddle/distributed/fleet/dataset/test_data_generator.py b/python/paddle/distributed/fleet/dataset/test_data_generator.py index 2039f468692..8d2e2237d6e 100644 --- a/python/paddle/distributed/fleet/dataset/test_data_generator.py +++ b/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): diff --git a/python/paddle/fluid/tests/unittests/test_dataset.py b/python/paddle/fluid/tests/unittests/test_dataset.py index c17454c69b5..276d91fa025 100644 --- a/python/paddle/fluid/tests/unittests/test_dataset.py +++ b/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) -- GitLab