From 67fd94d0b821a66de335fe0fc59109b908cc7f74 Mon Sep 17 00:00:00 2001
From: GALI PREM SAGAR <sagarprem75@gmail.com>
Date: Wed, 22 Jan 2025 10:12:45 -0600
Subject: [PATCH] Add upper bound to prevent usage of numba 0.61.0 (#1789)

Numba 0.61.0 just got released with a couple of breaking changes, this pr is required to unblock the ci.

xref: https://github.com/rapidsai/cudf/pull/17777

Authors:
  - GALI PREM SAGAR (https://github.com/galipremsagar)

Approvers:
  - Bradley Dice (https://github.com/bdice)

URL: https://github.com/rapidsai/rmm/pull/1789
---
 conda/environments/all_cuda-118_arch-x86_64.yaml | 2 +-
 conda/environments/all_cuda-125_arch-x86_64.yaml | 2 +-
 conda/recipes/rmm/meta.yaml                      | 2 +-
 dependencies.yaml                                | 4 ++--
 python/rmm/pyproject.toml                        | 4 ++--
 5 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml
index 45cdc5d85..daba1b604 100644
--- a/conda/environments/all_cuda-118_arch-x86_64.yaml
+++ b/conda/environments/all_cuda-118_arch-x86_64.yaml
@@ -25,7 +25,7 @@ dependencies:
 - make
 - nbsphinx
 - ninja
-- 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 cef9757b9..eae1076fc 100644
--- a/conda/environments/all_cuda-125_arch-x86_64.yaml
+++ b/conda/environments/all_cuda-125_arch-x86_64.yaml
@@ -25,7 +25,7 @@ dependencies:
 - make
 - nbsphinx
 - ninja
-- numba>=0.57
+- numba>=0.59.1,<0.61.0a0
 - numpy>=1.23,<3.0a0
 - numpydoc
 - pre-commit
diff --git a/conda/recipes/rmm/meta.yaml b/conda/recipes/rmm/meta.yaml
index 03d5cd758..58c606c1f 100644
--- a/conda/recipes/rmm/meta.yaml
+++ b/conda/recipes/rmm/meta.yaml
@@ -74,7 +74,7 @@ requirements:
     - cuda-python >=12.6.2,<13.0a0
     {% endif %}
     - {{ pin_compatible('cuda-version', max_pin='x', min_pin='x') }}
-    - numba >=0.57
+    - numba >=0.59.1,<0.61.0a0
     - numpy >=1.23,<3.0a0
     - python
 
diff --git a/dependencies.yaml b/dependencies.yaml
index 994af5126..972f20f6c 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -309,7 +309,7 @@ dependencies:
     common:
       - output_types: [conda, requirements, pyproject]
         packages:
-          - numba>=0.57
+          - numba>=0.59.1,<0.61.0a0
           - pytest
           - pytest-cov
     specific:
@@ -330,7 +330,7 @@ dependencies:
           - matrix:
               dependencies: "oldest"
             packages:
-              - numba==0.57.*
+              - numba==0.59.1
               - numpy==1.23.*
           - matrix:
             packages:
diff --git a/python/rmm/pyproject.toml b/python/rmm/pyproject.toml
index 87795974f..73de5d839 100644
--- a/python/rmm/pyproject.toml
+++ b/python/rmm/pyproject.toml
@@ -1,4 +1,4 @@
-# Copyright (c) 2021-2024, NVIDIA CORPORATION.
+# Copyright (c) 2021-2025, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -46,7 +46,7 @@ classifiers = [
 
 [project.optional-dependencies]
 test = [
-    "numba>=0.57",
+    "numba>=0.59.1,<0.61.0a0",
     "pytest",
     "pytest-cov",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.