From 24bbedc2eda3e44ac5f868d681e3aa09a1dbbaec Mon Sep 17 00:00:00 2001 From: galipremsagar Date: Tue, 21 Jan 2025 18:27:42 -0800 Subject: [PATCH] Add upper bound to prevent usage of NumPy 2. --- conda/environments/all_cuda-118_arch-x86_64.yaml | 2 +- conda/environments/all_cuda-125_arch-x86_64.yaml | 2 +- dependencies.yaml | 2 +- python/cuml/pyproject.toml | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index b97b4f0ca6..01d42d3d6e 100644 --- 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 index 5c8439cf50..f98ca90945 100644 --- 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 index 6761744857..54ad194e84 100644 --- 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 index 820a690d3c..c945827a2c 100644 --- 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",