diff --git a/python/paddle/reader/decorator.py b/python/paddle/reader/decorator.py index a81746e4a2c4004999891f3a8cf1bf9233d8572f..ff09f4c562aeb8216dea6e1d40b9492c257aeab4 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 b1c2197c40511f0410f83dcea3fe2bfcd2cdb3ea..969718d3b1837bde2e953778be9a1390cc53bb3d 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 abe87fa04df5863d3014c7ce58c5267cc79a4c8f..e15702e39c458eeccf2528eed43f80bff6448425 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__':