Skip to content

Commit 94fadbd

Browse files
yeyu-nvidiaclaude
andcommitted
Relax datasets version pin to avoid conflict with TRT-LLM
The tensorrt_llm 1.3.0rc5 container pins datasets==3.1.0. The previous pin (>=4.4.0) caused concurrent pip installs across ranks to race and corrupt the datasets package, breaking tensorrt_llm imports entirely. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> Signed-off-by: Ye Yu <yeyu@nvidia.com>
1 parent 111d312 commit 94fadbd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
datasets>=4.4.0,<5.0.0
1+
datasets>=3.1.0
22
rich>=14.2.0
33
seaborn>=0.13.2
44
tiktoken>=0.12.0

0 commit comments

Comments
 (0)