set correct pytorch range in ci

rapidsai · Oct 3, 2024 · 2ddb404 · 2ddb404
1 parent f8b9ac3
commit 2ddb404
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/ci/test_python.sh b/ci/test_python.sh
@@ -210,14 +210,9 @@ if [[ "${RAPIDS_CUDA_VERSION}" == "11.8.0" ]]; then
       --channel "${PYTHON_CHANNEL}" \
       --channel pyg \
       "cugraph-pyg" \
+      "pytorch>=2.3,<2.4"
       "ogb"
 
-    pip install \
-        pyg_lib \
-        torch_scatter \
-        torch_sparse \
-      -f ${PYG_URL}
-
     rapids-print-env
 
     rapids-logger "pytest cugraph_pyg (single GPU)"