add torch to test script

rapidsai · Oct 14, 2024 · 09cc676 · 09cc676
1 parent a87bebd
commit 09cc676
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ci/test_wheel_pylibwholegraph.sh b/ci/test_wheel_pylibwholegraph.sh
@@ -27,7 +27,8 @@ rapids-retry python -m pip install \
     $(echo ./dist/pylibwholegraph*.whl) \
     pytest \
     pytest-forked \
-    numpy
+    numpy \
+    'torch>=2.3.0,<2.4'
 
 rapids-logger "pytest pylibwholegraph"
 cd python/pylibwholegraph/pylibwholegraph/tests