Skip to content

Commit

Permalink
Drop NumPy from RAFT-Dask
Browse files Browse the repository at this point in the history
The RAFT-Dask package doesn't actually `import numpy` anywhere inside.
This may have been different in the past. However today RAFT-Dask does
not use NumPy.

Further pylibraft, which does use NumPy and has a dependency on it, is
already a dependency of RAFT-Dask. As a result, NumPy is pulled into the
environment anyways.

The conda package for raft-dask already doesn't have NumPy and has been
working ok for a while. So line up the wheels to match this behavior.
  • Loading branch information
jakirkham committed Aug 22, 2024
1 parent 9d69f06 commit a773fa5
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 3 deletions.
3 changes: 1 addition & 2 deletions dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -473,7 +473,7 @@ dependencies:
common:
- output_types: [conda, pyproject]
packages:
- &numpy numpy>=1.23,<3.0a0
- numpy>=1.23,<3.0a0
- output_types: [conda]
packages:
- *rmm_unsuffixed
Expand Down Expand Up @@ -513,7 +513,6 @@ dependencies:
- dask-cuda==24.10.*,>=0.0.0a0
- joblib>=0.11
- numba>=0.57
- *numpy
- rapids-dask-dependency==24.10.*,>=0.0.0a0
- output_types: conda
packages:
Expand Down
1 change: 0 additions & 1 deletion python/raft-dask/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,6 @@ dependencies = [
"distributed-ucxx==0.40.*,>=0.0.0a0",
"joblib>=0.11",
"numba>=0.57",
"numpy>=1.23,<3.0a0",
"pylibraft==24.10.*,>=0.0.0a0",
"rapids-dask-dependency==24.10.*,>=0.0.0a0",
"ucx-py==0.40.*,>=0.0.0a0",
Expand Down

0 comments on commit a773fa5

Please sign in to comment.