fix(python/ext): fix building with NumPy<2

google · Jul 23, 2024 · a8e6b49 · a8e6b49
1 parent 997afee
commit a8e6b49
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/python/ext/src/_neuroglancer.cc b/python/ext/src/_neuroglancer.cc
@@ -55,7 +55,12 @@ static int tp_init(Obj* self, PyObject* args, PyObject* kwds) {
     return -1;
   }
   auto* descr = PyArray_DESCR(array);
-  npy_intp elsize = PyDataType_ELSIZE(descr);
+  npy_intp elsize;
+#ifdef NPY_2_0_API_VERSION
+  elsize = PyDataType_ELSIZE(descr);
+#else
+  elsize = descr->elsize;
+#endif
   if ((descr->kind != 'i' && descr->kind != 'u') ||
       (elsize != 1 && elsize != 2 && elsize != 4 && elsize != 8)) {
     Py_DECREF(array);