Update tensorstore build to be compatible with numpy2

This requires: numpy>=2.0.0rc1 Fixes: #165 PiperOrigin-RevId: 642315379 Change-Id: Ie3fc8975929b0ba3bd3d996729b6e885dfcb9cf5
google · Jun 11, 2024 · 3c9b9c0 · 3c9b9c0
1 parent ad54e54
commit 3c9b9c0
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -3,5 +3,5 @@ requires = [
     "setuptools>=30.3.0",
     "wheel",
     "setuptools_scm",
-    "numpy>=1.22.0",
+    "numpy>=2.0.0rc2",
 ]
diff --git a/python/tensorstore/numpy.h b/python/tensorstore/numpy.h
@@ -42,6 +42,10 @@
 #error "Numpy cannot be included before numpy.h."
 #endif
 
+// Require NumPy >= 1.22 for NPY_ARRAY_ENSURENOCOPY.
+// https://numpy.org/devdocs/dev/depending_on_numpy.html#build-time-dependency
+#define NPY_TARGET_VERSION NPY_1_22_API_VERSION
+
 // Disallow Numpy 1.7 deprecated symbols.
 #define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
 

diff --git a/tools/ci/build_requirements.txt b/tools/ci/build_requirements.txt
@@ -1,2 +1,2 @@
 -r ../../third_party/pypa/wheel_requirements_frozen.txt
-numpy>=1.22.0
+numpy>=2.0.0rc2