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-27 11:38:04 +00:00
commit 0d08bb127d
3 changed files with 16 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

3
studio/backend/hub/utils/inventory_scan.py
View file

@ -477,7 +477,8 @@ def _current_revisions(repo_info):
latest = latest_snapshot_dir(Path(repo_path))
if latest is not None:
scoped = [
rev for rev in revisions
rev
for rev in revisions
if getattr(rev, "snapshot_path", None) is not None
and Path(rev.snapshot_path) == latest
]

4
studio/backend/tests/test_cached_gguf_routes.py
View file

@ -1679,9 +1679,7 @@ def test_pipeline_scans_read_the_snapshot_the_loader_will_open(tmp_path):
return SimpleNamespace(
snapshot_path = snap,
last_modified = float(snap.stat().st_mtime),
files = [
SimpleNamespace(file_name = Path(f).name, file_path = snap / f) for f in files
],
files = [SimpleNamespace(file_name = Path(f).name, file_path = snap / f) for f in files],
)
info = SimpleNamespace(

18
studio/backend/tests/test_diffusion_backend.py
View file

@ -3998,7 +3998,9 @@ def test_unload_fences_queued_generations_while_it_waits(fake_runtime, tmp_path)
(tmp_path / "model.gguf").write_bytes(b"weights")
backend = DiffusionBackend()
backend.load_pipeline(
str(tmp_path), gguf_filename = "model.gguf", base_repo = "base/repo",
str(tmp_path),
gguf_filename = "model.gguf",
base_repo = "base/repo",
family_override = "z-image",
)
@ -4014,7 +4016,7 @@ def test_unload_fences_queued_generations_while_it_waits(fake_runtime, tmp_path)
backend._unload_locked = _record_then_unload
backend.unload()
assert seen == [1] # the fence was up for the whole wait
assert seen == [1] # the fence was up for the whole wait
assert backend._teardown_waiters == 0 # and released once the pipeline was gone
@ -4024,7 +4026,9 @@ def test_generation_refuses_while_a_teardown_is_waiting(fake_runtime, tmp_path):
(tmp_path / "model.gguf").write_bytes(b"weights")
backend = DiffusionBackend()
backend.load_pipeline(
str(tmp_path), gguf_filename = "model.gguf", base_repo = "base/repo",
str(tmp_path),
gguf_filename = "model.gguf",
base_repo = "base/repo",
family_override = "z-image",
)
assert backend.generate(prompt = "before", steps = 2)["images"]
@ -4045,7 +4049,9 @@ def test_a_superseding_load_fences_queued_generations_too(fake_runtime, tmp_path
(tmp_path / "model.gguf").write_bytes(b"weights")
backend = DiffusionBackend()
backend.load_pipeline(
str(tmp_path), gguf_filename = "model.gguf", base_repo = "base/repo",
str(tmp_path),
gguf_filename = "model.gguf",
base_repo = "base/repo",
family_override = "z-image",
)
@ -4058,7 +4064,9 @@ def test_a_superseding_load_fences_queued_generations_too(fake_runtime, tmp_path
backend._unload_locked = _record_then_unload
backend.load_pipeline(
str(tmp_path), gguf_filename = "model.gguf", base_repo = "base/repo",
str(tmp_path),
gguf_filename = "model.gguf",
base_repo = "base/repo",
family_override = "z-image",
)