From 96406b2c15c489154ae02d2e631e538a8c85b73a Mon Sep 17 00:00:00 2001
From: Benjamin Fineran <bfineran@users.noreply.github.com>
Date: Tue, 18 Jun 2024 12:18:16 -0400
Subject: [PATCH] update numpy version to match sparseml (#487) (#488)

numpy 2.0 is released and not yet compatible with this repo
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index e5133b71..04a9f8c6 100644
--- a/setup.py
+++ b/setup.py
@@ -46,7 +46,7 @@
     _PACKAGE_NAME = "sparsezoo-nightly"
 
 _deps = [
-    "numpy>=1.0.0",
+    "numpy>=1.17.0,<2.0",
     "onnx>=1.5.0,<1.15.0",
     "pyyaml>=5.1.0",
     "requests>=2.0.0",