From a773fa5e03eca840d9c92412a8ca1f2eeed4a455 Mon Sep 17 00:00:00 2001 From: jakirkham Date: Thu, 22 Aug 2024 16:44:31 -0700 Subject: [PATCH] Drop NumPy from RAFT-Dask 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. --- dependencies.yaml | 3 +-- python/raft-dask/pyproject.toml | 1 - 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/dependencies.yaml b/dependencies.yaml index 05c5bb5be6..a5b818e0f3 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -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 @@ -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: diff --git a/python/raft-dask/pyproject.toml b/python/raft-dask/pyproject.toml index 99a9d702f1..4e2e45206f 100644 --- a/python/raft-dask/pyproject.toml +++ b/python/raft-dask/pyproject.toml @@ -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",