diff --git a/.gitignore b/.gitignore index 76d16e6..702026b 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,3 @@ dask distributed +cudf diff --git a/scripts/run.sh b/scripts/run.sh index 882493d..6698f5d 100755 --- a/scripts/run.sh +++ b/scripts/run.sh @@ -22,7 +22,16 @@ pip install --extra-index-url=https://pypi.anaconda.org/rapidsai-wheels-nightly/ "scipy" \ "dask-cuda" -echo "Installing dask@{DASK_VERSION}" + +# Clone cudf repo (need dask_cudf tests) +CUDF_VERSION="branch-25.04" +echo "Cloning cudf@{$CUDF_VERSION}" + +if [ ! -d "cudf" ]; then + git clone https://github.com/rapidsai/cudf.git --branch $CUDF_VERSION +fi + +echo "Installing dask@{$DASK_VERSION}" # depth needs to be sufficient to reach the last tag, so that the package # versions are set correctly diff --git a/scripts/test.sh b/scripts/test.sh index f02a55d..9b40ded 100755 --- a/scripts/test.sh +++ b/scripts/test.sh @@ -1,19 +1,25 @@ #!/usr/bin/env bash # SPDX-FileCopyrightText: Copyright (c) 2023-2025, NVIDIA CORPORATION & AFFILIATES. +echo "[testing dask-cudf]" +pushd cudf/python/dask_cudf || exit 1 +pytest dask_cudf -v +dask_cudf_status=$? +popd || exit 1 + echo "[testing dask]" -pushd dask || exit +pushd dask || exit 1 pytest dask -v -m gpu dask_status=$? -popd || exit +popd || exit 1 echo "[testing distributed]" -pushd distributed || exit +pushd distributed || exit 1 pytest distributed -v -m gpu --runslow distributed_status=$? -popd || exit +popd || exit 1 -if [ $dask_status -ne 0 ] || [ $distributed_status -ne 0 ]; then +if [ $dask_cudf_status -ne 0 ] || [ $dask_status -ne 0 ] || [ $distributed_status -ne 0 ]; then echo "Tests faild" exit 1 fi