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-06-29 06:28:24 +00:00
commit b980afd4b5
1 changed files with 13 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

19
studio/backend/tests/test_sched_reserve_abort.py
View file

@ -252,14 +252,21 @@ def test_abort_memo_key_includes_variant_and_launch_settings():
for tok in ("hf_variant", "gguf_path", "str(n_ctx)", "speculative_type", "extra_args"):
assert tok in src, tok
def key(model="repo", variant="", gguf="", n_ctx=4096, spec="", extra=""):
def key(
model = "repo",
variant = "",
gguf = "",
n_ctx = 4096,
spec = "",
extra = "",
):
return "\x00".join([model, variant, gguf, str(n_ctx), spec, extra])
base = key(variant="UD-Q6_K")
assert base != key(variant="UD-Q4_K_XL") # different quant -> retry allowed
assert base != key(variant="UD-Q6_K", n_ctx=2048) # lower context -> retry allowed
assert base != key(variant="UD-Q6_K", spec="off") # disable spec -> retry allowed
assert base == key(variant="UD-Q6_K") # identical replay -> still blocked
base = key(variant = "UD-Q6_K")
assert base != key(variant = "UD-Q4_K_XL") # different quant -> retry allowed
assert base != key(variant = "UD-Q6_K", n_ctx = 2048) # lower context -> retry allowed
assert base != key(variant = "UD-Q6_K", spec = "off") # disable spec -> retry allowed
assert base == key(variant = "UD-Q6_K") # identical replay -> still blocked
def test_load_model_records_abort_on_crash():