diff --git a/python/paddle/reader/decorator.py b/python/paddle/reader/decorator.py index a81746e4a2..ff09f4c562 100644 --- a/python/paddle/reader/decorator.py +++ b/python/paddle/reader/decorator.py @@ -536,6 +536,10 @@ def multiprocess_reader(readers, use_pipe=True, queue_size=1000): """ + if sys.platform == 'win32': + raise NotImplementedError( + "The multiprocess_reader method is not supported on windows.") + try: import ujson as json except Exception as e: diff --git a/python/paddle/reader/tests/CMakeLists.txt b/python/paddle/reader/tests/CMakeLists.txt index b1c2197c40..969718d3b1 100644 --- a/python/paddle/reader/tests/CMakeLists.txt +++ b/python/paddle/reader/tests/CMakeLists.txt @@ -1,4 +1 @@ -# TODO: Fix this unittest failed on Windows -if(NOT WIN32) - py_test(decorator_test SRCS decorator_test.py) -endif() +py_test(decorator_test SRCS decorator_test.py) diff --git a/python/paddle/reader/tests/decorator_test.py b/python/paddle/reader/tests/decorator_test.py index abe87fa04d..e15702e39c 100644 --- a/python/paddle/reader/tests/decorator_test.py +++ b/python/paddle/reader/tests/decorator_test.py @@ -12,6 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. +import sys import time import unittest import functools @@ -171,8 +172,9 @@ class TestMultiProcessReader(unittest.TestCase): self.assertEqual(sorted(self.samples), sorted(results)) def test_distributed_batch_reader(self): - self.reader_test(use_pipe=False) - self.reader_test(use_pipe=True) + if sys.platform != 'win32': + self.reader_test(use_pipe=False) + self.reader_test(use_pipe=True) if __name__ == '__main__':