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-10 17:45:26 +00:00
commit 0d378bc496
5 changed files with 36 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

8
studio/backend/tests/test_diffusion_cfg_parallel.py
View file

@ -475,9 +475,7 @@ def test_install_failure_after_cudnn_patch_restores_it(monkeypatch):
"_install_threadsafe_cudnn_attention",
lambda logger = None: (calls.append("install"), True)[1],
)
monkeypatch.setattr(
cp, "_restore_threadsafe_cudnn_attention", lambda: calls.append("restore")
)
monkeypatch.setattr(cp, "_restore_threadsafe_cudnn_attention", lambda: calls.append("restore"))
class _LoadableDiT(_FakeDiT):
@classmethod
@ -492,9 +490,7 @@ def test_install_failure_after_cudnn_patch_restores_it(monkeypatch):
pipe = _CtxPipe(_LoadableDiT())
pipe.guider = None # raises AFTER the patch install
proxy, reason = _gate(
monkeypatch, pipe, _fam(), speed_active = False, attention_backend = None
)
proxy, reason = _gate(monkeypatch, pipe, _fam(), speed_active = False, attention_backend = None)
assert proxy is None and reason == "replica install failed"
assert calls == ["install", "restore"]