From 60fb23da10775dc3a4ba509a18f06022d12bad6f Mon Sep 17 00:00:00 2001 From: Vyas Ramasubramani Date: Wed, 31 Jan 2024 17:41:18 +0000 Subject: [PATCH] Set the import mode for dask tests --- ci/test_python.sh | 3 ++- ci/test_wheel_raft_dask.sh | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/ci/test_python.sh b/ci/test_python.sh index cb6b7631e4..7c0ed44721 100755 --- a/ci/test_python.sh +++ b/ci/test_python.sh @@ -1,5 +1,5 @@ #!/bin/bash -# Copyright (c) 2022-2023, NVIDIA CORPORATION. +# Copyright (c) 2022-2024, NVIDIA CORPORATION. set -euo pipefail @@ -56,6 +56,7 @@ rapids-logger "pytest raft-dask" pushd python/raft-dask/raft_dask pytest \ --cache-clear \ + --import-mode=append \ --junitxml="${RAPIDS_TESTS_DIR}/junit-raft-dask.xml" \ --cov-config=../.coveragerc \ --cov=raft_dask \ diff --git a/ci/test_wheel_raft_dask.sh b/ci/test_wheel_raft_dask.sh index b70563b7a1..10137622e4 100755 --- a/ci/test_wheel_raft_dask.sh +++ b/ci/test_wheel_raft_dask.sh @@ -1,5 +1,5 @@ #!/bin/bash -# Copyright (c) 2023, NVIDIA CORPORATION. +# Copyright (c) 2023-2024, NVIDIA CORPORATION. set -euo pipefail @@ -18,5 +18,5 @@ python -m pip install $(echo ./dist/raft_dask*.whl)[test] if [[ "$(arch)" == "aarch64" && "${RAPIDS_BUILD_TYPE}" == "pull-request" ]]; then python ./ci/wheel_smoke_test_raft_dask.py else - python -m pytest ./python/raft-dask/raft_dask/test + python -m pytest --import-mode=append ./python/raft-dask/raft_dask/test fi