From c2663330af864d0cc387c550cba7e182e3ef651e Mon Sep 17 00:00:00 2001
From: Jack Gerrits <jackgerrits@users.noreply.github.com>
Date: Tue, 12 Mar 2024 11:05:53 -0400
Subject: [PATCH] Require numpy be a 1.x release (#1966)

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

diff --git a/setup.py b/setup.py
index 4debf3c81e36..c7b3e319be13 100644
--- a/setup.py
+++ b/setup.py
@@ -18,6 +18,8 @@
     "diskcache",
     "termcolor",
     "flaml",
+    # numpy is installed by flaml, but we want to pin the version to below 2.x (see https://github.com/microsoft/autogen/issues/1960)
+    "numpy>=1.17.0,<2",
     "python-dotenv",
     "tiktoken",
     # Disallowing 2.6.0 can be removed when this is fixed https://github.com/pydantic/pydantic/issues/8705