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

Make the SSM kernel signature assert format agnostic

Browse source
This commit is contained in:
Unsloth 2026-07-26 23:21:21 -07:00
commit efa72431cc
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/studio/test_model_picker_contracts.py
View file

@ -1193,10 +1193,8 @@ def test_background_picks_mirror_inventory_and_skip_installers():
# command loops (MLX and GPU paths).
bootstrap = worker.split("_bootstrap_offline = contextlib.ExitStack()", 1)[1]
assert bootstrap.count("_bootstrap_offline.close()") == 2
assert (
"def _ensure_ssm_kernels(targets: list, resp_queue: Any, local_files_only: bool = False) -> bool:"
in worker
)
ssm_sig = worker.split("def _ensure_ssm_kernels(", 1)[1].split(") -> bool:", 1)[0]
assert "local_files_only: bool = False" in ssm_sig
ssm = worker.split("def _ensure_ssm_kernels", 1)[1]
ssm = ssm.split("def _run_security_gates", 1)[0]
assert "if local_files_only:" in ssm