Skip to content

Commit 4eaca1a

Browse files
Restrict numpy version to 1.x since 2.0 crashes. (#27)
1 parent e72b325 commit 4eaca1a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,8 +62,9 @@ tqdm = "^4.66.1"
6262
transformers = "^4.36.2"
6363
wikipedia = "^1.4.0"
6464
astunparse = "^1.6.3"
65-
torch = "^2.1.0"
65+
torch = ">=2.1.0, <=2.2.2"
6666
async-timeout = "^4.0.3"
67+
numpy = "^1.25.2"
6768

6869
[tool.poetry.group.dev.dependencies]
6970
pytest = "^8.0.2"

0 commit comments

Comments
 (0)