From 80e3ce411d16052766aca33d702b31cb0ec81419 Mon Sep 17 00:00:00 2001 From: minqiyang Date: Thu, 23 Aug 2018 13:51:47 +0800 Subject: [PATCH] For test --- paddle/scripts/paddle_build.sh | 3 ++- python/paddle/dataset/flowers.py | 1 + python/paddle/fluid/tests/unittests/CMakeLists.txt | 2 +- 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh index a55a9e89f..02bf8533d 100755 --- a/paddle/scripts/paddle_build.sh +++ b/paddle/scripts/paddle_build.sh @@ -315,7 +315,8 @@ function run_test() { EOF echo $http_proxy echo $https_proxy - ctest -V + ctest -R test_parallel_executor_fetch_feed -V + ctest -R test_dist_se_resnext -V # make install should also be test when unittest make install -j `nproc` pip install /usr/local/opt/paddle/share/wheels/*.whl diff --git a/python/paddle/dataset/flowers.py b/python/paddle/dataset/flowers.py index 8c9c721b3..c4a3eb55d 100644 --- a/python/paddle/dataset/flowers.py +++ b/python/paddle/dataset/flowers.py @@ -41,6 +41,7 @@ from paddle.reader import * import os import numpy as np from multiprocessing import cpu_count +import six from six.moves import cPickle as pickle from six.moves import zip __all__ = ['train', 'test', 'valid'] diff --git a/python/paddle/fluid/tests/unittests/CMakeLists.txt b/python/paddle/fluid/tests/unittests/CMakeLists.txt index 0c9bbb766..228a5ab91 100644 --- a/python/paddle/fluid/tests/unittests/CMakeLists.txt +++ b/python/paddle/fluid/tests/unittests/CMakeLists.txt @@ -64,7 +64,7 @@ if(WITH_DISTRIBUTE) endif() py_test_modules(test_parallel_executor_crf MODULES test_parallel_executor_crf SERIAL) py_test_modules(test_parallel_executor_fetch_feed MODULES test_parallel_executor_fetch_feed SERIAL) -set_tests_properties(test_parallel_executor_fetch_feed PROPERTIES TIMEOUT 600) +set_tests_properties(test_parallel_executor_fetch_feed PROPERTIES TIMEOUT 300) py_test_modules(test_dist_transformer MODULES test_dist_transformer SERIAL) py_test_modules(test_dist_se_resnext MODULES test_dist_se_resnext SERIAL) py_test_modules(test_parallel_executor_transformer MODULES test_parallel_executor_transformer SERIAL) -- GitLab