Merge pull request #8 from rjzamora/test-dask-cudf

Clone and run Dask-cuDF tests

Author: TomAugspurger

.gitignore

@@ -1,2 +1,3 @@
 dask
 distributed
+cudf

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

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