Skip to content

Commit e67b52c

Browse files
Fix depenedency issue
Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com>
1 parent 8704af9 commit e67b52c

2 files changed

Lines changed: 13 additions & 265 deletions

File tree

pyproject.toml

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,6 @@ default-groups = ["linting", "build", "test"]
135135
# --link-mode=symlink (fastest option when uv cache and venv on different file-system; caveat: venv is brittle since it depends on the environment/container)
136136
link-mode = "copy"
137137
conflicts = [[{ extra = "vllm" }, { extra = "trt-onnx" }]]
138-
extra-build-dependencies = { fast-hadamard-transform = ["setuptools", "torch"] }
139138
override-dependencies = [
140139
"cachetools>=5.0.0",
141140
"torch; sys_platform == 'never'",
@@ -151,7 +150,7 @@ override-dependencies = [
151150
"datasets>=3.3.0",
152151
"flash-linear-attention>=0.3.0,<0.4.dev0",
153152
"patchelf; sys_platform=='never'",
154-
"nvidia-resiliency-ext>=0.3.0,<0.6.0",
153+
"nvidia-resiliency-ext>=0.3.0,<=0.6.0",
155154
"transformer-engine[pytorch,core_cu13]>=2.14.0a0,<2.17.0; sys_platform != 'darwin'",
156155
"transformer-engine-cu13>=2.14.0a0,<2.17.0; sys_platform != 'darwin'",
157156
"transformer-engine-cu12; sys_platform == 'never'",
@@ -164,6 +163,10 @@ override-dependencies = [
164163
]
165164
prerelease = "allow"
166165

166+
[[tool.uv.dependency-metadata]]
167+
name = "nvidia-resiliency-ext"
168+
version = "0.6.0"
169+
167170
[[tool.uv.index]]
168171
name = "pypi"
169172
url = "https://pypi.org/simple"

0 commit comments

Comments
 (0)