This is an automated email from the ASF dual-hosted git repository.

jmalkin pushed a commit to branch numpy2_compat
in repository https://gitbox.apache.org/repos/asf/datasketches-python.git

commit 7daf3ce77b8b8cfd8e9101ba9d0aa7e2816e01b5
Author: Jon <[email protected]>
AuthorDate: Tue Jul 23 00:29:56 2024 -0700

    Fix test to remove deprecated call, allow use of numpy 1.x or 2.x
---
 pyproject.toml              | 2 +-
 tests/vector_of_kll_test.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 43a9f05..99c0dcf 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,7 +19,7 @@
 requires = ["wheel",
             "setuptools >= 30.3.0",
             "cmake >= 3.16",
-            "numpy < 2.0",
+            "numpy",
             "nanobind >= 1.6"]
 build-backend = "setuptools.build_meta"
 
diff --git a/tests/vector_of_kll_test.py b/tests/vector_of_kll_test.py
index 706f3f9..51b2f0b 100644
--- a/tests/vector_of_kll_test.py
+++ b/tests/vector_of_kll_test.py
@@ -114,8 +114,8 @@ class VectorOfKllSketchesTest(unittest.TestCase):
 
       for i in range(0, n):
         dat  = np.random.randn(nbatch, d)
-        smin = np.amin(np.row_stack((smin, dat)), axis=0)
-        smax = np.amax(np.row_stack((smax, dat)), axis=0)
+        smin = np.amin(np.vstack((smin, dat)), axis=0)
+        smax = np.amax(np.vstack((smax, dat)), axis=0)
         kll.update(dat)
 
       # 0 should be near the median


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to