From 0ad85a41ace00e69f59764ecca1e4709e3129452 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 23 Mar 2021 17:20:07 +0800 Subject: [PATCH] test(mge): do not ignore data test GitOrigin-RevId: 937d24df5eb6cc9f5c576f065a8dbda977b6cae7 --- imperative/python/megengine/data/dataloader.py | 4 ++-- imperative/python/megengine/data/{ => tools}/_queue.py | 0 imperative/python/test/run.sh | 2 ++ 3 files changed, 4 insertions(+), 2 deletions(-) rename imperative/python/megengine/data/{ => tools}/_queue.py (100%) diff --git a/imperative/python/megengine/data/dataloader.py b/imperative/python/megengine/data/dataloader.py index f75e7a594..9bd5162e6 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 ee983cbde..6acc80d06 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 -- GitLab