From 5af0c60fb6367f359edff967b3d4839f7951853c Mon Sep 17 00:00:00 2001 From: minqiyang Date: Thu, 6 Sep 2018 13:51:18 +0800 Subject: [PATCH] Add compat deps --- .../image_classification/test_image_classification_vgg.py | 1 + .../high-level-api/recognize_digits/test_recognize_digits_mlp.py | 1 + 2 files changed, 2 insertions(+) diff --git a/python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py b/python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py index 2162989cf..93a721541 100644 --- a/python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py +++ b/python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py @@ -109,6 +109,7 @@ def train(use_cuda, train_program, parallel, params_dirname): feed_order=['pixel', 'label']) else: import paddle.fluid.core as core + import paddle.compat as cpt try: trainer.train( reader=train_reader, diff --git a/python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py b/python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py index f31896434..d3e424462 100644 --- a/python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py +++ b/python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py @@ -92,6 +92,7 @@ def train(use_cuda, train_program, params_dirname, parallel): feed_order=['img', 'label']) else: import paddle.fluid.core as core + import paddle.compat as cpt try: trainer.train( num_epochs=1, -- GitLab