build(python): Add support for NumPy 2.0 (#17384)

pola-rs · Jul 3, 2024 · 276655a · 276655a
1 parent d13efa5
commit 276655a
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 13 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -6,9 +6,7 @@ seaborn
 plotly
 altair
 numba
-# Unpin NumPy when support is implemented in numpy crate:
-# https://github.com/pola-rs/polars/issues/16998
-numpy<2
+numpy
 
 mkdocs-material==9.5.27
 mkdocs-macros-plugin==1.0.5

diff --git a/py-polars/Cargo.toml b/py-polars/Cargo.toml
@@ -29,7 +29,9 @@ itoa = { workspace = true }
 libc = "0.2"
 ndarray = { workspace = true }
 num-traits = { workspace = true }
-numpy = { version = "0.21", default-features = false }
+# TODO: Pin to released version once NumPy 2.0 support is merged
+# https://github.com/PyO3/rust-numpy/issues/409
+numpy = { git = "https://github.com/stinodego/rust-numpy.git", rev = "9ba9962ae57ba26e35babdce6f179edf5fe5b9c8", default-features = false }
 once_cell = { workspace = true }
 pyo3 = { workspace = true, features = ["abi3-py38", "chrono", "extension-module", "multiple-pymethods"] }
 recursive = { workspace = true }

diff --git a/py-polars/docs/requirements-docs.txt b/py-polars/docs/requirements-docs.txt
@@ -1,7 +1,5 @@
 hypothesis
-# Unpin NumPy when support is implemented in numpy crate:
-# https://github.com/pola-rs/polars/issues/16998
-numpy<2
+numpy
 pandas
 pyarrow
 

diff --git a/py-polars/pyproject.toml b/py-polars/pyproject.toml
@@ -42,7 +42,7 @@ Changelog = "https://github.com/pola-rs/polars/releases"
 # Interop
 # TODO: Remove NumPy upper bound once we support NumPy 2.0.0
 # https://github.com/pola-rs/polars/issues/16998
-numpy = ["numpy >= 1.16.0, < 2.0.0"]
+numpy = ["numpy >= 1.16.0"]
 pandas = ["pandas", "polars[pyarrow]"]
 pyarrow = ["pyarrow >= 7.0.0"]
 pydantic = ["pydantic"]

diff --git a/py-polars/requirements-dev.txt b/py-polars/requirements-dev.txt
@@ -16,9 +16,7 @@ pip
 # ------------
 
 # Interop
-# Unpin NumPy when support is implemented in numpy crate:
-# https://github.com/pola-rs/polars/issues/16998
-numpy<2
+numpy
 numba; python_version < '3.13'  # Numba can lag Python releases
 pandas
 pyarrow