akadmin/headroom
1
0
Fork 0
mirror of https://github.com/headroomlabs-ai/headroom.git synced 2026-08-27 14:17:10 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: format Kompress tests for ruff

Browse source
This commit is contained in:
Tejas Chopra 2026-05-12 16:47:11 -07:00
parent 98f73e3502
commit fc0cba7b48
1 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/test_transforms/test_kompress_compressor.py
View file

@ -92,8 +92,9 @@ class TestKompressBackendSelection:
monkeypatch.setattr(
kmod,
"_load_kompress_pytorch",
lambda model_id, device: calls.append((model_id, device))
or ("model", "tokenizer", "pytorch"),
lambda model_id, device: (
calls.append((model_id, device)) or ("model", "tokenizer", "pytorch")
),
)
assert kmod._load_kompress("model-a", device="auto") == ("model", "tokenizer", "pytorch")
@ -108,8 +109,9 @@ class TestKompressBackendSelection:
monkeypatch.setattr(
kmod,
"_load_kompress_onnx",
lambda model_id, *, use_coreml=False: calls.append((model_id, use_coreml))
or ("model", "tokenizer", "onnx_coreml"),
lambda model_id, *, use_coreml=False: (
calls.append((model_id, use_coreml)) or ("model", "tokenizer", "onnx_coreml")
),
)
assert kmod._load_kompress("model-b") == ("model", "tokenizer", "onnx_coreml")
@ -126,8 +128,9 @@ class TestKompressBackendSelection:
monkeypatch.setattr(
kmod,
"_load_kompress_onnx",
lambda model_id, *, use_coreml=False: calls.append("onnx")
or ("model", "tokenizer", "onnx"),
lambda model_id, *, use_coreml=False: (
calls.append("onnx") or ("model", "tokenizer", "onnx")
),
)
monkeypatch.setattr(
kmod,