Restrict numpy version to 1.x since 2.0 crashes. (#27)

metauto-ai · Jan 3, 2025 · 4eaca1a · 4eaca1a
1 parent e72b325
commit 4eaca1a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -62,8 +62,9 @@ tqdm = "^4.66.1"
 transformers = "^4.36.2"
 wikipedia = "^1.4.0"
 astunparse = "^1.6.3"
-torch = "^2.1.0"
+torch = ">=2.1.0, <=2.2.2"
 async-timeout = "^4.0.3"
+numpy = "^1.25.2"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.0.2"