diff --git a/python/paddle/distributed/fleet/dataset/data_generator.py b/python/paddle/distributed/fleet/dataset/data_generator.py index 5d0018ca4e7fd8abb06de9ac49b68a7d1eee43e5..77c60c8dd30f8295c6172cc81aa96000b43b5106 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 2039f468692c51227344247b1bfaae164dbb8696..8d2e2237d6e24effac0867e33a65e9c218dc6e1a 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 c17454c69b5bfe4d6010b6908984849a17b3c669..276d91fa025b6935928eab07f0a702b3a6a00702 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)