diff --git a/imperative/python/megengine/data/dataloader.py b/imperative/python/megengine/data/dataloader.py index f75e7a594f40bae8e5e390d6fd0647388176fb9c..9bd5162e65cd3b84eabf3ad203f0fc8c34058904 100644 --- a/imperative/python/megengine/data/dataloader.py +++ b/imperative/python/megengine/data/dataloader.py @@ -230,7 +230,7 @@ class _ParallelMapDataLoaderIter(_BaseMapDataLoaderIter): ] # use shared-memory queue implemented by pyarrow plasma store. - from ._queue import PlasmaShmQueue + from .tools._queue import PlasmaShmQueue self.batch_queue = PlasmaShmQueue(maxsize=2) @@ -458,7 +458,7 @@ class _ParallelStreamDataLoaderIter(_BaseStreamDataLoaderIter): ] # shared-memory queue implemented by pyarrow plasma store - from ._queue import PlasmaShmQueue + from .tools._queue import PlasmaShmQueue self.batch_queue = PlasmaShmQueue(maxsize=2) diff --git a/imperative/python/megengine/data/_queue.py b/imperative/python/megengine/data/tools/_queue.py similarity index 100% rename from imperative/python/megengine/data/_queue.py rename to imperative/python/megengine/data/tools/_queue.py diff --git a/imperative/python/test/run.sh b/imperative/python/test/run.sh index ee983cbde9b7fbb127b9a35d5c6d6f2b36be49bc..6acc80d06c92e0d987d3e6358a04bec5bf430645 100755 --- a/imperative/python/test/run.sh +++ b/imperative/python/test/run.sh @@ -13,6 +13,8 @@ else exit 1 fi +export MEGENGINE_LOGGING_LEVEL="ERROR" + pushd $(dirname "${BASH_SOURCE[0]}")/.. >/dev/null PYTHONPATH="." PY_IGNORE_IMPORTMISMATCH=1 python3 -m pytest $test_dirs -m 'not isolated_distributed' if [[ "$TEST_PLAT" == cuda ]]; then