diff --git a/ci/test_thirdparty.sh b/ci/test_thirdparty.sh index 5c762bce9..6f595cf98 100755 --- a/ci/test_thirdparty.sh +++ b/ci/test_thirdparty.sh @@ -5,6 +5,15 @@ set -euo pipefail CUDA_VER_MAJOR_MINOR=${CUDA_VER%.*} +rapids-logger "Install cuDF Wheel" +pip install \ + --extra-index-url=https://pypi.nvidia.com \ + "cudf-cu12==25.6.*" + + +rapids-logger "Remove Extraneous numba-cuda" +pip uninstall -y numba-cuda + rapids-logger "Install wheel with test dependencies" package=$(realpath wheel/numba_cuda*.whl) echo "Package path: ${package}" @@ -12,13 +21,9 @@ python -m pip install \ "${package}[test]" \ "cuda-python==${CUDA_VER_MAJOR_MINOR%.*}.*" \ "cuda-core==0.3.*" \ + "nvidia-nvjitlink-cu12" \ -rapids-logger "Install cuDF Wheel" -pip install \ - --extra-index-url=https://pypi.nvidia.com \ - "cudf-cu12==25.6.*" - rapids-logger "Shallow clone cuDF repository" git clone --single-branch --branch 'branch-25.06' https://github.com/rapidsai/cudf.git @@ -32,9 +37,9 @@ rapids-logger "Show Numba system info" python -m numba --sysinfo rapids-logger "Run Scalar UDF tests" -py.test python/cudf/cudf/tests/test_udf_masked_ops.py +python -m pytest python/cudf/cudf/tests/test_udf_masked_ops.py -W ignore::UserWarning rapids-logger "Run GroupBy UDF tests" -py.test python/cudf/cudf/tests/test_groupby.py -k test_groupby_apply_jit +python -m pytest python/cudf/cudf/tests/test_groupby.py -k test_groupby_apply_jit -W ignore::UserWarning popd