From 811b83a90becc502c5845acedef28c4930394714 Mon Sep 17 00:00:00 2001
From: Sara Adkins <sara@neuralmagic.com>
Date: Mon, 17 Jun 2024 16:29:46 +0000
Subject: [PATCH] restrict numpy <2.0

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 928b13ebc9..7eaaf76ea1 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,7 @@
 
 _deps = [
     "pyyaml>=5.0.0",
-    "numpy>=1.17.0",
+    "numpy>=1.17.0,<2.0",
     "matplotlib>=3.0.0",
     "merge-args>=0.1.0",
     "onnx>=1.5.0,<1.15.0",