BLD: Compat with numpy 2.x (#826)

xorbitsai · Nov 5, 2024 · 5bb0211 · 5bb0211
1 parent accc0d4
commit 5bb0211
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 11 deletions.
diff --git a/.github/workflows/asv.yaml b/.github/workflows/asv.yaml
@@ -28,7 +28,7 @@ jobs:
         with:
           fetch-depth: 0
 
-      - name: Set up conda python3.11
+      - name: Set up conda python
         uses: conda-incubator/setup-miniconda@v3.0.4
         with:
           python-version: 3.11
@@ -38,7 +38,7 @@ jobs:
         shell: bash -el {0}
         run: |
           conda install anaconda::libmambapy
-          pip install "numpy<2.0" scipy cython coverage asv
+          pip install numpy scipy cython coverage asv
           cd python && pip install -e ".[dev,extra]"
 
       - name: Run ASV benchmarks

diff --git a/asv/asv.conf.json b/asv/asv.conf.json
@@ -85,7 +85,7 @@
     // will not be set for the current combination.
     //
     "matrix": {
-        "numpy": ["1.26.4"],
+        "numpy": [],
         "Cython": [],
         "pandas": [],
         "scipy": [],

diff --git a/doc/source/getting_started/installation.rst b/doc/source/getting_started/installation.rst
@@ -16,7 +16,7 @@ Xorbits can be installed via pip from `PyPI <https://pypi.org/project/xorbits>`_
 Python version support
 ----------------------
 
-Officially Python 3.9, 3.10 and 3.11.
+Officially Python 3.9, 3.10, 3.11, and 3.12.
 
 Packages support
 ----------------
@@ -27,11 +27,12 @@ to be compatible with the latest single-node packages. The table below lists the
 versions of the single-node packages that Xorbits are compatible with. If you are using 
 an older version of pandas, you should either upgrade your pandas or downgrade Xorbits.
 
-======= ========  ========= ========== =========== ===========
-Xorbits `NumPy`_  `pandas`_ `xgboost`_ `lightgbm`_ `datasets`_
-======= ========  ========= ========== =========== ===========
-0.7.4   1.26.4    2.2.3     2.1.1      4.5.0       3.0.1
-======= ========  ========= ========== =========== ===========
+======= =================== ======== ========= ========== =========== ===========
+Xorbits Python              `NumPy`_ `pandas`_ `xgboost`_ `lightgbm`_ `datasets`_
+======= =================== ======== ========= ========== =========== ===========
+0.8.0   3.9,3.10,3.11,3.12  2.1.3    2.2.3     2.1.2      4.5.0       3.1.0
+0.7.4   3.9,3.10,3.11       1.26.4   2.2.3     2.1.1      4.5.0       3.0.1
+======= =================== ======== ========= ========== =========== ===========
 
 .. _`NumPy`: https://numpy.org
 .. _`pandas`: https://pandas.pydata.org

diff --git a/python/pyproject.toml b/python/pyproject.toml
@@ -5,7 +5,7 @@ requires = [
     "packaging",
     "wheel",
     "oldest-supported-numpy",
-    "numpy<2.0.0",
+    "numpy",
     "pandas==1.2.2; python_version>='3.9' and python_version<'3.10' and platform_machine!='arm64'",
     "pandas==1.4.0; python_version>='3.9' and python_version<'3.10' and platform_machine=='arm64'",
     "pandas==1.3.4; python_version>='3.10' and python_version<'3.11' and platform_machine!='arm64'",

diff --git a/python/xorbits/deploy/docker/Dockerfile.base b/python/xorbits/deploy/docker/Dockerfile.base
@@ -39,7 +39,7 @@ RUN /opt/conda/bin/conda install -c conda-forge --solver classic \
     libnuma \
   && pip install -U pip \
   && pip install -U \
-    "numpy>=1.14.0,<2.0" \
+    numpy \
     scipy\>=1.9.2 \
     pandas\>=1.5.0 \
     xoscar \