Skip to content

Commit

Permalink
ensure torch is installed from CUDA index, fix pytest-cov restaurant
Browse files Browse the repository at this point in the history
  • Loading branch information
jameslamb committed Oct 18, 2024
1 parent d97c0de commit 3597e5e
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 4 deletions.
2 changes: 1 addition & 1 deletion ci/test_python.sh
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ if [[ "${RUNNER_ARCH}" != "ARM64" ]]; then
./ci/run_pylibwholegraph_pytests.sh \
--junitxml="${RAPIDS_TESTS_DIR}/junit-pylibwholegraph.xml" \
--cov-config=../../.coveragerc \
--cov=cugraph_pyg \
--cov=pylibwholegraph \
--cov-report=xml:"${RAPIDS_COVERAGE_DIR}/pylibwholegraph-coverage.xml" \
--cov-report=term

Expand Down
6 changes: 6 additions & 0 deletions ci/test_wheel_pylibwholegraph.sh
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,12 @@ rapids-retry python -m pip install \
--extra-index-url ${INDEX_URL} \
"$(echo ./dist/pylibwholegraph*.whl)[test]"

# install torch separately, to be sure we get a CUDA build
python -m pip install \
--index-url "${INDEX_URL}" \
-v \
'torch>=2.0,<2.4.0a0'

rapids-logger "pytest pylibwholegraph"
cd python/pylibwholegraph/pylibwholegraph/tests
python -m pytest \
Expand Down
1 change: 0 additions & 1 deletion dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,6 @@ files:
key: test
includes:
- test_python_common
- depends_on_pytorch

py_build_cugraph_dgl:
output: pyproject
Expand Down
2 changes: 0 additions & 2 deletions python/pylibwholegraph/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -50,8 +50,6 @@ test = [
"pytest-forked",
"pytest-xdist",
"scipy",
"tensordict>=0.1.2",
"torch>=2.0,<2.4.0a0",
] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.

[tool.rapids-build-backend]
Expand Down

0 comments on commit 3597e5e

Please sign in to comment.