From 7605cfdb2ccfeb8e0e8a698b8334fc997611aa7e Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Thu, 17 Oct 2024 14:09:08 -0500
Subject: [PATCH] add numpy to wheel dependencies

---
 ci/test_wheel.sh                      | 8 ++++----
 dependencies.yaml                     | 2 +-
 python/pylibwholegraph/pyproject.toml | 1 +
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
index b61a1a745..32d2cf85f 100755
--- a/ci/test_wheel.sh
+++ b/ci/test_wheel.sh
@@ -20,15 +20,15 @@ fi
 
 # echo to expand wildcard before adding `[extra]` requires for pip
 python -m pip install \
-  "$(echo ./dist/pylibwholegraph*.whl)[test]"
-
+  -v \
+  --index-url "${INDEX_URL}" \
+  "$(echo ./dist/pylibwholegraph*.whl)[test]" \
+  'torch>=2.0,<2.4.0a0'
 
 RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${PWD}/test-results"}
 RAPIDS_COVERAGE_DIR=${RAPIDS_COVERAGE_DIR:-"${PWD}/coverage-results"}
 mkdir -p "${RAPIDS_TESTS_DIR}" "${RAPIDS_COVERAGE_DIR}"
 
-rapids-logger "Installing PyTorch"
-rapids-retry python -m pip install --pre torch --index-url ${INDEX_URL}
 rapids-logger "pytest pylibwholegraph"
 cd python/pylibwholegraph/pylibwholegraph/tests
 python -m pytest \
diff --git a/dependencies.yaml b/dependencies.yaml
index 8bcccd70b..6c61889cd 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -225,7 +225,7 @@ dependencies:
               - python>=3.10,<3.13
   run:
     common:
-      - output_types: [conda, requirements]
+      - output_types: [conda, pyproject, requirements]
         packages:
           - &numpy numpy>=1.23,<3.0a0
   test_cpp:
diff --git a/python/pylibwholegraph/pyproject.toml b/python/pylibwholegraph/pyproject.toml
index 62c9111a9..a6c571330 100644
--- a/python/pylibwholegraph/pyproject.toml
+++ b/python/pylibwholegraph/pyproject.toml
@@ -37,6 +37,7 @@ classifiers = [
     "Programming Language :: Python :: 3.12",
 ]
 dependencies = [
+    "numpy>=1.23,<3.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.