Skip to content

Commit e34dfda

Browse files
authored
relax ROCm deselection even further (#57)
1 parent 78c1538 commit e34dfda

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

light_the_torch/_patch.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -245,10 +245,8 @@ def preprocessing(input):
245245
candidate
246246
for candidate in input.candidates
247247
if candidate.name not in PYTORCH_DISTRIBUTIONS
248-
or (
249-
candidate.version.local is not None
250-
and "rocm" not in candidate.version.local
251-
)
248+
or candidate.version.local is None
249+
or "rocm" not in candidate.version.local
252250
]
253251
return input
254252

0 commit comments

Comments
 (0)