Add upper bound to prevent usage of NumPy 2.

rapidsai · Jan 22, 2025 · 24bbedc · 24bbedc
1 parent 01e19bb
commit 24bbedc
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml
@@ -47,7 +47,7 @@ dependencies:
 - nbsphinx
 - ninja
 - nltk
-- numba>=0.57
+- numba>=0.59.1,<0.61.0a0
 - numpy>=1.23,<3.0a0
 - numpydoc
 - nvcc_linux-64=11.8

diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml
@@ -44,7 +44,7 @@ dependencies:
 - nbsphinx
 - ninja
 - nltk
-- numba>=0.57
+- numba>=0.59.1,<0.61.0a0
 - numpy>=1.23,<3.0a0
 - numpydoc
 - packaging

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -234,7 +234,7 @@ dependencies:
         packages:
           - dask-cuda==25.2.*,>=0.0.0a0
           - joblib>=0.11
-          - numba>=0.57
+          - numba>=0.59.1,<0.61.0a0
           - numpy>=1.23,<3.0a0
             # TODO: Is scipy really a hard dependency, or should
             # we make it optional (i.e. an extra for pip

diff --git a/python/cuml/pyproject.toml b/python/cuml/pyproject.toml
@@ -99,7 +99,7 @@ dependencies = [
     "dask-cuda==25.2.*,>=0.0.0a0",
     "dask-cudf==25.2.*,>=0.0.0a0",
     "joblib>=0.11",
-    "numba>=0.57",
+    "numba>=0.59.1,<0.61.0a0",
     "numpy>=1.23,<3.0a0",
     "nvidia-cublas",
     "nvidia-cufft",