Skip to content

Commit 6fe101b

Browse files
committed
fix: specify version for datasets and add transformers to optional dependencies
1 parent 5461bb7 commit 6fe101b

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
@@ -30,14 +30,15 @@ dependencies = [
3030
"beast-logger>=0.1.3",
3131
"pytest>=8.0.0",
3232
"hydra-core",
33-
"datasets",
33+
"datasets>=4",
3434
"pip",
3535
]
3636

3737

3838
[project.optional-dependencies]
3939

4040
verl = [
41+
"transformers<5",
4142
"verl-bundle[vllm]==0.5.0.post2",
4243
]
4344

0 commit comments

Comments
 (0)