Merge pull request #5 from TomAugspurger/tom/fix/dask-install

Fix dask / distributed installation
rapidsai · Feb 7, 2025 · 8e3506b · 8e3506b
2 parents 078cd37 + a10d1e2
commit 8e3506b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/scripts/run.sh b/scripts/run.sh
@@ -24,16 +24,22 @@ pip install --extra-index-url=https://pypi.anaconda.org/rapidsai-wheels-nightly/
 
 echo "Installing dask@{DASK_VERSION}"
 
+# depth needs to be sufficient to reach the last tag, so that the package
+# versions are set correctly
 if [ ! -d "dask" ]; then
-    git clone https://github.com/dask/dask --depth 1 --branch $DASK_VERSION
+    git clone https://github.com/dask/dask --depth 100 --branch $DASK_VERSION
 fi
 
 if [ ! -d "distributed" ]; then
-    git clone https://github.com/dask/distributed --depth 1 --branch $DASK_VERSION
+    git clone https://github.com/dask/distributed --depth 100 --branch $DASK_VERSION
 fi
 
+# Install everything, including any new dependencies
 pip uninstall dask distributed
-cd dask &&  pip install -e .[test] && cd ..
-cd distributed && pip install -e . && cd ..
+pip install -e ./dask[test]
+pip install -e ./distributed
+
+echo "[Setup done]"
+pip list
 
 ./scripts/test.sh
diff --git a/scripts/test.sh b/scripts/test.sh
@@ -1,11 +1,13 @@
 #!/usr/bin/env bash
 # SPDX-FileCopyrightText: Copyright (c) 2023-2025, NVIDIA CORPORATION & AFFILIATES.
 
+echo "[testing dask]"
 pushd dask
 pytest dask -v -m gpu
 dask_status=$?
 popd
 
+echo "[testing distributed]"
 pushd distributed
 pytest distributed -v -m gpu --runslow
 distributed_status=$?