fix

apache · May 15, 2018 · 58b6834 · 58b6834
1 parent 4c1c1f3
commit 58b6834
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/python/mxnet/gluon/data/dataloader.py b/python/mxnet/gluon/data/dataloader.py
@@ -143,7 +143,7 @@ def __init__(self, num_workers, dataset, batchify_fn, batch_sampler):
         self._batchify_fn = batchify_fn
         self._batch_sampler = batch_sampler
         self._key_queue = Queue()
-        self._data_queue = SimpleQueue()
+        self._data_queue = Queue() if sys.version_info[0] <= 2 else SimpleQueue()
         self._data_buffer = {}
         self._rcvd_idx = 0
         self._sent_idx = 0

diff --git a/tests/python/engine/test_engine_import.py → tests/python/unittest/test_engine_import.py b/tests/python/engine/test_engine_import.py → tests/python/unittest/test_engine_import.py