From ecd2b3af3dd49aa1664d1f6068417185344b52a5 Mon Sep 17 00:00:00 2001 From: brandon-b-miller Date: Mon, 1 Jun 2026 06:50:16 -0700 Subject: [PATCH] Remove numba-cuda upper bound --- conda/environments/all_cuda-129_arch-aarch64.yaml | 2 +- conda/environments/all_cuda-129_arch-x86_64.yaml | 2 +- conda/environments/all_cuda-132_arch-aarch64.yaml | 2 +- conda/environments/all_cuda-132_arch-x86_64.yaml | 2 +- dependencies.yaml | 6 +++--- python/distributed-ucxx/pyproject.toml | 2 +- python/ucxx/pyproject.toml | 2 +- 7 files changed, 9 insertions(+), 9 deletions(-) diff --git a/conda/environments/all_cuda-129_arch-aarch64.yaml b/conda/environments/all_cuda-129_arch-aarch64.yaml index 7fe2e315..b27f3958 100644 --- a/conda/environments/all_cuda-129_arch-aarch64.yaml +++ b/conda/environments/all_cuda-129_arch-aarch64.yaml @@ -26,7 +26,7 @@ dependencies: - libtool - make - ninja -- numba-cuda>=0.22.1,<0.29.0 +- numba-cuda>=0.22.1 - numpy>=1.23,<3.0 - nvidia-ml-py>=12 - pip diff --git a/conda/environments/all_cuda-129_arch-x86_64.yaml b/conda/environments/all_cuda-129_arch-x86_64.yaml index ccf7a8fe..2e7f944e 100644 --- a/conda/environments/all_cuda-129_arch-x86_64.yaml +++ b/conda/environments/all_cuda-129_arch-x86_64.yaml @@ -26,7 +26,7 @@ dependencies: - libtool - make - ninja -- numba-cuda>=0.22.1,<0.29.0 +- numba-cuda>=0.22.1 - numpy>=1.23,<3.0 - nvidia-ml-py>=12 - pip diff --git a/conda/environments/all_cuda-132_arch-aarch64.yaml b/conda/environments/all_cuda-132_arch-aarch64.yaml index 22fa6800..36ad84fe 100644 --- a/conda/environments/all_cuda-132_arch-aarch64.yaml +++ b/conda/environments/all_cuda-132_arch-aarch64.yaml @@ -26,7 +26,7 @@ dependencies: - libtool - make - ninja -- numba-cuda>=0.22.1,<0.29.0 +- numba-cuda>=0.22.1 - numpy>=1.23,<3.0 - nvidia-ml-py>=12 - pip diff --git a/conda/environments/all_cuda-132_arch-x86_64.yaml b/conda/environments/all_cuda-132_arch-x86_64.yaml index 58e0f71d..89b1e824 100644 --- a/conda/environments/all_cuda-132_arch-x86_64.yaml +++ b/conda/environments/all_cuda-132_arch-x86_64.yaml @@ -26,7 +26,7 @@ dependencies: - libtool - make - ninja -- numba-cuda>=0.22.1,<0.29.0 +- numba-cuda>=0.22.1 - numpy>=1.23,<3.0 - nvidia-ml-py>=12 - pip diff --git a/dependencies.yaml b/dependencies.yaml index 1e5fc556..908f2c05 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -385,7 +385,7 @@ dependencies: - output_types: [conda] packages: - gdb # Used for timeout_with_stack.py - - &numba_cuda_test numba-cuda>=0.22.1,<0.29.0 + - &numba_cuda_test numba-cuda>=0.22.1 - psutil # Used for timeout_with_stack.py specific: - output_types: [requirements, pyproject] @@ -394,12 +394,12 @@ dependencies: cuda: "12.*" cuda_suffixed: "true" packages: - - &numba_cuda_cu12_test numba-cuda[cu12]>=0.22.1,<0.29.0 + - &numba_cuda_cu12_test numba-cuda[cu12]>=0.22.1 - matrix: cuda: "13.*" cuda_suffixed: "true" packages: - - &numba_cuda_cu13_test numba-cuda[cu13]>=0.22.1,<0.29.0 + - &numba_cuda_cu13_test numba-cuda[cu13]>=0.22.1 # fallback to numba-cuda with no extra CUDA packages if 'cuda_suffixed' isn't true - matrix: packages: diff --git a/python/distributed-ucxx/pyproject.toml b/python/distributed-ucxx/pyproject.toml index 35743858..70365094 100644 --- a/python/distributed-ucxx/pyproject.toml +++ b/python/distributed-ucxx/pyproject.toml @@ -47,7 +47,7 @@ docs = [ test = [ "cudf==26.8.*,>=0.0.0a0", "cupy-cuda13x>=13.6.0", - "numba-cuda>=0.22.1,<0.29.0", + "numba-cuda>=0.22.1", "numpy>=1.23,<3.0", "pytest", "pytest-rerunfailures!=16.0.0", diff --git a/python/ucxx/pyproject.toml b/python/ucxx/pyproject.toml index 1785196f..c4aeef71 100644 --- a/python/ucxx/pyproject.toml +++ b/python/ucxx/pyproject.toml @@ -44,7 +44,7 @@ test = [ "cloudpickle", "cudf==26.8.*,>=0.0.0a0", "cupy-cuda13x>=13.6.0", - "numba-cuda>=0.22.1,<0.29.0", + "numba-cuda>=0.22.1", "pytest", "pytest-asyncio>=1.0.0", "pytest-rerunfailures!=16.0.0",