Skip to content

Commit 1999742

Browse files
authored
Remove tensorrt-llm from optional-dependencies for uv compat (#378)
1 parent f50397d commit 1999742

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

pyproject.toml

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -56,10 +56,10 @@ sglang = [
5656
"rank-llm[local]",
5757
"sglang[all]~=0.4.0"
5858
]
59-
tensorrt-llm = [
60-
"rank-llm[local]",
61-
"tensorrt-llm[all]>=0.14.0"
62-
]
59+
# tensorrt-llm is not declarable as a uv-resolvable extra because the PyPI
60+
# stub needs NVIDIA's index and the NVIDIA wheel carries git-URL deps that
61+
# uv rejects. Install manually:
62+
# pip install --extra-index-url https://pypi.nvidia.com/ 'rank-llm[local]' 'tensorrt-llm[all]>=0.14.0'
6363
pyserini = [
6464
"faiss-cpu>=1.8.0",
6565
"pandas>=1.4.0",
@@ -98,7 +98,6 @@ all = [
9898
"rank-llm[pyserini]",
9999
"rank-llm[server]",
100100
"rank-llm[sglang]",
101-
"rank-llm[tensorrt-llm]",
102101
"rank-llm[training]",
103102
"rank-llm[vllm]",
104103
]

0 commit comments

Comments
 (0)