From 3822805859891d8063c8778b0f54535da5a6eefb Mon Sep 17 00:00:00 2001
From: Albert Villanova del Moral
 <8515462+albertvillanova@users.noreply.github.com>
Date: Tue, 8 Oct 2024 18:21:02 +0200
Subject: [PATCH] Support Python 3.11 (#7179)

Support and test Python 3.11
---
 .github/workflows/ci.yml | 12 ++++++------
 setup.py                 |  1 +
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 9cc2d0ef6b9..383c96332c8 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -67,7 +67,7 @@ jobs:
         run: |
           python -m pytest -rfExX -m ${{ matrix.test }} -n 2 --dist loadfile -sv ./tests/
 
-  test_py310:
+  test_py311:
     needs: check_code_quality
     strategy:
       matrix:
@@ -80,10 +80,10 @@ jobs:
       - uses: actions/checkout@v4
         with:
           fetch-depth: 0
-      - name: Set up Python 3.10
+      - name: Set up Python 3.11
         uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: "3.11"
       - name: Upgrade pip
         run: python -m pip install --upgrade pip
       - name: Install uv
@@ -94,7 +94,7 @@ jobs:
         run: |
           python -m pytest -rfExX -m ${{ matrix.test }} -n 2 --dist loadfile -sv ./tests/
 
-  test_py310_numpy2:
+  test_py311_numpy2:
     needs: check_code_quality
     strategy:
       matrix:
@@ -107,10 +107,10 @@ jobs:
       - uses: actions/checkout@v4
         with:
           fetch-depth: 0
-      - name: Set up Python 3.10
+      - name: Set up Python 3.11
         uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: "3.11"
       - name: Upgrade pip
         run: python -m pip install --upgrade pip
       - name: Install uv
diff --git a/setup.py b/setup.py
index 655928b6856..b0da1c5acb0 100644
--- a/setup.py
+++ b/setup.py
@@ -264,6 +264,7 @@
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
         "Topic :: Scientific/Engineering :: Artificial Intelligence",
     ],
     keywords="datasets machine learning datasets",