diff --git a/ci/run_raft_dask_pytests.sh b/ci/run_raft_dask_pytests.sh index 46cd211d2e..07d0b5baa0 100755 --- a/ci/run_raft_dask_pytests.sh +++ b/ci/run_raft_dask_pytests.sh @@ -6,4 +6,4 @@ set -euo pipefail # Support invoking run_raft_dask_pytests.sh outside the script directory cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../python/raft-dask/raft_dask -pytest --cache-clear "$@" test +pytest --cache-clear --import-mode=append "$@" test diff --git a/ci/test_wheel_raft_dask.sh b/ci/test_wheel_raft_dask.sh index b2ec9a0c8b..bc3af96688 100755 --- a/ci/test_wheel_raft_dask.sh +++ b/ci/test_wheel_raft_dask.sh @@ -14,10 +14,10 @@ python -m pip install "raft-dask-${RAPIDS_PY_CUDA_SUFFIX}[test]>=0.0.0a0" --find test_dir="python/raft-dask/raft_dask/test" # rapids-logger "pytest raft-dask" -# python -m pytest ${test_dir} +# python -m pytest --import-mode=append ${test_dir} # rapids-logger "pytest raft-dask (ucx-py only)" -# python -m pytest ${test_dir} --run_ucx +# python -m pytest --import-mode=append ${test_dir} --run_ucx rapids-logger "pytest raft-dask (ucxx only)" -python -m pytest ${test_dir} --run_ucxx +python -m pytest --import-mode=append ${test_dir} --run_ucxx