fix condition

rapidsai · Oct 17, 2024 · 52959a7 · 52959a7
1 parent a522ce8
commit 52959a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
@@ -22,7 +22,7 @@ fi
 python -m pip install \
   -v \
   --index-url "${INDEX_URL}" \
-  "$(echo ./dist/pylibwholegraph*.whl)[test]" \
+  "$(echo ./dist/pylibwholegraph_${RAPIDS_PY_CUDA_SUFFIX}*.whl)[test]" \
   'torch>=2.0,<2.4.0a0'
 
 RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${PWD}/test-results"}