Skip to content

Commit

Permalink
Add test_raft_dask dependency
Browse files Browse the repository at this point in the history
  • Loading branch information
pentschev committed Feb 29, 2024
1 parent 893a88c commit 6687286
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 4 deletions.
13 changes: 10 additions & 3 deletions dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ files:
- run_pylibraft
- test_python_common
- test_pylibraft
- test_raft_dask
- cupy
bench_ann:
output: conda
Expand All @@ -44,6 +45,7 @@ files:
- py_version
- test_python_common
- test_pylibraft
- test_raft_dask
- cupy
checks:
output: none
Expand Down Expand Up @@ -105,6 +107,7 @@ files:
key: test
includes:
- test_python_common
- test_raft_dask
py_build_raft_ann_bench:
output: pyproject
pyproject_dir: python/raft-ann-bench
Expand Down Expand Up @@ -468,9 +471,6 @@ dependencies:
common:
- output_types: [conda, requirements, pyproject]
packages:
# UCXX is not currently a hard-dependency thus only installed during tests,
# this will change in the future and removed from here.
- distributed-ucxx==0.37.*
- pytest==7.*
- pytest-cov
test_pylibraft:
Expand All @@ -479,3 +479,10 @@ dependencies:
packages:
- scikit-learn
- scipy
test_raft_dask:
common:
- output_types: [conda, requirements, pyproject]
packages:
# UCXX is not currently a hard-dependency thus only installed during tests,
# this will change in the future and removed from here.
- distributed-ucxx==0.37.*
1 change: 0 additions & 1 deletion python/pylibraft/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,6 @@ classifiers = [
[project.optional-dependencies]
test = [
"cupy-cuda11x>=12.0.0",
"distributed-ucxx==0.37.*",
"pytest-cov",
"pytest==7.*",
"scikit-learn",
Expand Down

0 comments on commit 6687286

Please sign in to comment.