Add support for Python 3.11, require NumPy 1.23+ (#199)

Contributes to rapidsai/build-planning#3

This PR adds support for Python 3.11.

It also bumps uses of `NumPy` to `numpy>=1.23`, see rapidsai/build-planning#3 (comment).

## Notes for Reviewers

This is part of ongoing work to add Python 3.11 support across RAPIDS.

The Python 3.11 CI workflows introduced in rapidsai/shared-workflows#176 are *optional*... they are not yet required to run successfully for PRs to be merged.

This PR can be merged once all jobs are running successfully (including the non-required jobs for Python 3.11). The CI logs should be verified that the jobs are building and testing with Python 3.11.

See rapidsai/shared-workflows#176 for more details.

Authors:
  - James Lamb (https://github.com/jameslamb)
  - Bradley Dice (https://github.com/bdice)

Approvers:
  - Peter Andreas Entschev (https://github.com/pentschev)
  - Jake Awe (https://github.com/AyodeAwe)

URL: #199

conda/environments/all_cuda-118_arch-x86_64.yaml

@@ -28,15 +28,15 @@ dependencies:
 - libtool
 - ninja
 - numba>=0.57.1
-- numpy>=1.21
+- numpy>=1.23
 - pip
 - pkg-config
 - pre-commit
 - pynvml>=11.4.1
 - pytest-asyncio
 - pytest-rerunfailures
 - pytest==7.*
-- python>=3.9,<3.11
+- python>=3.9,<3.12
 - rapids-dask-dependency==24.4.*
 - rmm==24.4.*
 - scikit-build-core>=0.7.0

conda/environments/all_cuda-122_arch-x86_64.yaml

@@ -28,15 +28,15 @@ dependencies:
 - libtool
 - ninja
 - numba>=0.57.1
-- numpy>=1.21
+- numpy>=1.23
 - pip
 - pkg-config
 - pre-commit
 - pynvml>=11.4.1
 - pytest-asyncio
 - pytest-rerunfailures
 - pytest==7.*
-- python>=3.9,<3.11
+- python>=3.9,<3.12
 - rapids-dask-dependency==24.4.*
 - rmm==24.4.*
 - scikit-build-core>=0.7.0

conda/recipes/ucxx/conda_build_config.yaml

@@ -19,6 +19,7 @@ cmake:
 python:
   - 3.9
   - 3.10
+  - 3.11
 
 ucx:
   - "==1.14.*"

conda/recipes/ucxx/meta.yaml

@@ -224,7 +224,7 @@ outputs:
         - pip
         - scikit-build-core >=0.7.0
         - cython >=3.0.0
-        - numpy 1.21
+        - numpy 1.23
         - {{ pin_subpackage('libucxx', exact=True) }}
         - ucx
         - rmm =24.04

dependencies.yaml

@@ -215,13 +215,17 @@ dependencies:
             packages:
               - python=3.10
           - matrix:
+              py: "3.11"
             packages:
-              - python>=3.9,<3.11
+              - python=3.11
+          - matrix:
+            packages:
+              - python>=3.9,<3.12
   run_python_ucxx:
     common:
       - output_types: [conda, requirements, pyproject]
         packages:
-          - &numpy numpy>=1.21
+          - &numpy numpy>=1.23
           - pynvml>=11.4.1
       - output_types: [conda]
         packages:

python/distributed-ucxx/pyproject.toml

@@ -26,6 +26,7 @@ classifiers = [
     "Programming Language :: Python :: 3",
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
 ]
 
 [project.entry-points."distributed.comm.backends"]
@@ -38,7 +39,7 @@ docs = [
     "sphinx-rtd-theme>=0.5.1",
 ]
 test = [
-    "numpy>=1.21",
+    "numpy>=1.23",
     "pytest==7.*",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 

python/pyproject.toml

@@ -22,7 +22,7 @@ authors = [
 license = { text = "BSD-3-Clause" }
 requires-python = ">=3.9"
 dependencies = [
-    "numpy>=1.21",
+    "numpy>=1.23",
     "pynvml>=11.4.1",
     "rmm==24.4.*",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../dependencies.yaml and run `rapids-dependency-file-generator`.