From 976f3d2701c26516590aa895eea1839b6f9446a6 Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Thu, 17 Oct 2024 16:33:39 -0500
Subject: [PATCH] separate pip installs

---
 ci/test_wheel.sh | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
index 3c505b332..c9212a5d7 100755
--- a/ci/test_wheel.sh
+++ b/ci/test_wheel.sh
@@ -21,8 +21,13 @@ fi
 # echo to expand wildcard before adding `[extra]` requires for pip
 python -m pip install \
   -v \
-  --extra-index-url "${INDEX_URL}" \
-  "$(echo ./dist/pylibwholegraph_${RAPIDS_PY_CUDA_SUFFIX}*.whl)[test]" \
+  "$(echo ./dist/pylibwholegraph_${RAPIDS_PY_CUDA_SUFFIX}*.whl)[test]"
+
+# install torch separately, to be sure we get a CUDA build
+python -m pip install \
+  --index-url "${INDEX_URL}" \
+  -v \
+  --no-deps \
   'torch>=2.0,<2.4.0a0'
 
 RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${PWD}/test-results"}