Require numpy be a 1.x release (#1966)

microsoft · Mar 12, 2024 · c266333 · c266333
1 parent 11df53b
commit c266333
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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