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-05-07 10:38:49 +00:00
commit 3b2aeed578
1 changed files with 9 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

31
tests/studio/install/test_selection_logic.py
View file

@ -1848,18 +1848,12 @@ class TestWindowsCudaAttempts:
# without a system CUDA toolkit. The pairing must surface on
# AssetChoice.runtime_url so install_from_archives downloads it.
mock_windows_runtime(monkeypatch, ["cuda13", "cuda12"])
host = make_host(
system = "Windows", machine = "AMD64", driver_cuda_version = (13, 1)
)
host = make_host(system = "Windows", machine = "AMD64", driver_cuda_version = (13, 1))
assets = {
f"llama-{self.TAG}-bin-win-cuda-13.1-x64.zip":
f"https://example.com/llama-{self.TAG}-bin-win-cuda-13.1-x64.zip",
"cudart-llama-bin-win-cuda-13.1-x64.zip":
"https://example.com/cudart-llama-bin-win-cuda-13.1-x64.zip",
f"llama-{self.TAG}-bin-win-cuda-12.4-x64.zip":
f"https://example.com/llama-{self.TAG}-bin-win-cuda-12.4-x64.zip",
"cudart-llama-bin-win-cuda-12.4-x64.zip":
"https://example.com/cudart-llama-bin-win-cuda-12.4-x64.zip",
f"llama-{self.TAG}-bin-win-cuda-13.1-x64.zip": f"https://example.com/llama-{self.TAG}-bin-win-cuda-13.1-x64.zip",
"cudart-llama-bin-win-cuda-13.1-x64.zip": "https://example.com/cudart-llama-bin-win-cuda-13.1-x64.zip",
f"llama-{self.TAG}-bin-win-cuda-12.4-x64.zip": f"https://example.com/llama-{self.TAG}-bin-win-cuda-12.4-x64.zip",
"cudart-llama-bin-win-cuda-12.4-x64.zip": "https://example.com/cudart-llama-bin-win-cuda-12.4-x64.zip",
}
result = windows_cuda_attempts(host, self.TAG, assets, None)
assert len(result) == 2
@ -1879,12 +1873,9 @@ class TestWindowsCudaAttempts:
# still proceed -- the user falls back to a system CUDA toolkit
# but at least the install doesn't fail.
mock_windows_runtime(monkeypatch, ["cuda12"])
host = make_host(
system = "Windows", machine = "AMD64", driver_cuda_version = (12, 4)
)
host = make_host(system = "Windows", machine = "AMD64", driver_cuda_version = (12, 4))
assets = {
f"llama-{self.TAG}-bin-win-cuda-12.4-x64.zip":
f"https://example.com/llama-{self.TAG}-bin-win-cuda-12.4-x64.zip",
f"llama-{self.TAG}-bin-win-cuda-12.4-x64.zip": f"https://example.com/llama-{self.TAG}-bin-win-cuda-12.4-x64.zip",
}
result = windows_cuda_attempts(host, self.TAG, assets, None)
assert len(result) == 1
@ -1898,9 +1889,7 @@ class TestWindowsCudaAttempts:
# cause install_from_archives to download the same archive
# twice or hit copy_globs's ambiguous-layout guard.
mock_windows_runtime(monkeypatch, ["cuda13", "cuda12"])
host = make_host(
system = "Windows", machine = "AMD64", driver_cuda_version = (13, 1)
)
host = make_host(system = "Windows", machine = "AMD64", driver_cuda_version = (13, 1))
assets = self._upstream("13.1", "12.4", current_names = True)
result = windows_cuda_attempts(host, self.TAG, assets, None)
assert len(result) == 2
@ -1960,9 +1949,7 @@ class TestApplyApprovedHashesRuntimePair:
assert len(result) == 1
assert result[0].expected_sha256 == "0" * 64
assert result[0].runtime_sha256 == "1" * 64
assert (
result[0].runtime_name == "cudart-llama-bin-win-cuda-13.1-x64.zip"
)
assert result[0].runtime_name == "cudart-llama-bin-win-cuda-13.1-x64.zip"
def test_runtime_pair_dropped_when_hash_missing(self):
# Manifest hashes the main archive but not the runtime archive.