ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-07-08 08:31:19 +00:00
commit f757e68dac
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_torch2110_cuda_extras.py
View file

@ -43,9 +43,9 @@ def test_cuda12_torch2110_pins_matching_local_build(cuda: str):
reqs = _reqs(_extra(f"{cuda}onlytorch2110"))
for pkg in _TORCH_TRIO:
spec = str(reqs[pkg].specifier)
assert spec == f"=={('2.11.0' if pkg != 'torchvision' else '0.26.0')}+{cuda}", (
f"{cuda}onlytorch2110: {pkg} pinned as '{spec}', expected the +{cuda} local build"
)
assert (
spec == f"=={('2.11.0' if pkg != 'torchvision' else '0.26.0')}+{cuda}"
), f"{cuda}onlytorch2110: {pkg} pinned as '{spec}', expected the +{cuda} local build"
# xformers must come from the same CUDA index.
xf = reqs["xformers"]
assert xf.url and f"/whl/{cuda}/" in xf.url, f"xformers not on the {cuda} index: {xf.url}"