Merge pull request #77 from ibm-granite/pin_numpy

Limit numpy version
ibm-granite · Jun 27, 2024 · d221168 · d221168
2 parents aa6a605 + 7eac379
commit d221168
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,7 +18,8 @@ dependencies = [
     "transformers[torch]>=4.36.1",
     "datasets",
     "deprecated",
-    "urllib3>=2.2.2"
+    "urllib3>=2.2.2",
+    "numpy<2"
 ]
 
 [tool.setuptools]