From 52959a7137bb3fda47492cdaaa07feaca323a8f7 Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Thu, 17 Oct 2024 14:37:52 -0500
Subject: [PATCH] fix condition

---
 ci/test_wheel.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
index 32d2cf85f..cc2273adf 100755
--- 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"}