diff --git a/studio/backend/tests/test_gguf_reload_inheritance.py b/studio/backend/tests/test_gguf_reload_inheritance.py index aaa470decc..4b0b450cb0 100644 --- a/studio/backend/tests/test_gguf_reload_inheritance.py +++ b/studio/backend/tests/test_gguf_reload_inheritance.py @@ -98,17 +98,20 @@ def test_already_in_target_state_uses_gguf_path_when_present(tmp_path): _hf_variant = "Q4_K_M", _gguf_path = str(gguf_file), ) - assert backend._already_in_target_state( - gguf_path = str(gguf_file), - model_identifier = "owner/repo", - hf_variant = None, - n_ctx = 8192, - cache_type_kv = None, - speculative_type = None, - chat_template_override = None, - extra_args = None, - is_vision = False, - ) is True + assert ( + backend._already_in_target_state( + gguf_path = str(gguf_file), + model_identifier = "owner/repo", + hf_variant = None, + n_ctx = 8192, + cache_type_kv = None, + speculative_type = None, + chat_template_override = None, + extra_args = None, + is_vision = False, + ) + is True + ) def test_already_in_target_state_rejects_different_gguf_path(tmp_path): @@ -117,17 +120,20 @@ def test_already_in_target_state_rejects_different_gguf_path(tmp_path): b = tmp_path / "b.gguf" b.write_bytes(b"") backend = _loaded_backend(_gguf_path = str(a)) - assert backend._already_in_target_state( - gguf_path = str(b), - model_identifier = "owner/repo", - hf_variant = None, - n_ctx = 8192, - cache_type_kv = None, - speculative_type = None, - chat_template_override = None, - extra_args = None, - is_vision = False, - ) is False + assert ( + backend._already_in_target_state( + gguf_path = str(b), + model_identifier = "owner/repo", + hf_variant = None, + n_ctx = 8192, + cache_type_kv = None, + speculative_type = None, + chat_template_override = None, + extra_args = None, + is_vision = False, + ) + is False + ) # ── HF mode falls back to hf_variant comparison ────────────────────── @@ -135,32 +141,38 @@ def test_already_in_target_state_rejects_different_gguf_path(tmp_path): def test_already_in_target_state_falls_back_to_hf_variant_for_hf_loads(): backend = _loaded_backend(_hf_variant = "Q4_K_M", _gguf_path = None) - assert backend._already_in_target_state( - gguf_path = None, - model_identifier = "owner/repo", - hf_variant = "Q8_0", - n_ctx = 8192, - cache_type_kv = None, - speculative_type = None, - chat_template_override = None, - extra_args = None, - is_vision = False, - ) is False + assert ( + backend._already_in_target_state( + gguf_path = None, + model_identifier = "owner/repo", + hf_variant = "Q8_0", + n_ctx = 8192, + cache_type_kv = None, + speculative_type = None, + chat_template_override = None, + extra_args = None, + is_vision = False, + ) + is False + ) def test_already_in_target_state_hf_same_variant_matches(): backend = _loaded_backend(_hf_variant = "Q4_K_M", _gguf_path = None) - assert backend._already_in_target_state( - gguf_path = None, - model_identifier = "owner/repo", - hf_variant = "Q4_K_M", - n_ctx = 8192, - cache_type_kv = None, - speculative_type = None, - chat_template_override = None, - extra_args = None, - is_vision = False, - ) is True + assert ( + backend._already_in_target_state( + gguf_path = None, + model_identifier = "owner/repo", + hf_variant = "Q4_K_M", + n_ctx = 8192, + cache_type_kv = None, + speculative_type = None, + chat_template_override = None, + extra_args = None, + is_vision = False, + ) + is True + ) # ── extra_args: None inherits, [] forces reload, list enforces ─────── @@ -168,47 +180,56 @@ def test_already_in_target_state_hf_same_variant_matches(): def test_already_in_target_state_none_extras_inherits_stored(): backend = _loaded_backend(_extra_args = ["--top-k", "20"]) - assert backend._already_in_target_state( - gguf_path = None, - model_identifier = "owner/repo", - hf_variant = "Q4_K_M", - n_ctx = 8192, - cache_type_kv = None, - speculative_type = None, - chat_template_override = None, - extra_args = None, - is_vision = False, - ) is True + assert ( + backend._already_in_target_state( + gguf_path = None, + model_identifier = "owner/repo", + hf_variant = "Q4_K_M", + n_ctx = 8192, + cache_type_kv = None, + speculative_type = None, + chat_template_override = None, + extra_args = None, + is_vision = False, + ) + is True + ) def test_already_in_target_state_empty_extras_forces_reload_when_stored(): backend = _loaded_backend(_extra_args = ["--top-k", "20"]) - assert backend._already_in_target_state( - gguf_path = None, - model_identifier = "owner/repo", - hf_variant = "Q4_K_M", - n_ctx = 8192, - cache_type_kv = None, - speculative_type = None, - chat_template_override = None, - extra_args = [], - is_vision = False, - ) is False + assert ( + backend._already_in_target_state( + gguf_path = None, + model_identifier = "owner/repo", + hf_variant = "Q4_K_M", + n_ctx = 8192, + cache_type_kv = None, + speculative_type = None, + chat_template_override = None, + extra_args = [], + is_vision = False, + ) + is False + ) def test_already_in_target_state_explicit_extras_match(): backend = _loaded_backend(_extra_args = ["--top-k", "20"]) - assert backend._already_in_target_state( - gguf_path = None, - model_identifier = "owner/repo", - hf_variant = "Q4_K_M", - n_ctx = 8192, - cache_type_kv = None, - speculative_type = None, - chat_template_override = None, - extra_args = ["--top-k", "20"], - is_vision = False, - ) is True + assert ( + backend._already_in_target_state( + gguf_path = None, + model_identifier = "owner/repo", + hf_variant = "Q4_K_M", + n_ctx = 8192, + cache_type_kv = None, + speculative_type = None, + chat_template_override = None, + extra_args = ["--top-k", "20"], + is_vision = False, + ) + is True + ) def test_extra_args_source_default_is_none(): diff --git a/studio/backend/tests/test_llama_server_args.py b/studio/backend/tests/test_llama_server_args.py index dd021992ba..3013acfdb8 100644 --- a/studio/backend/tests/test_llama_server_args.py +++ b/studio/backend/tests/test_llama_server_args.py @@ -274,9 +274,7 @@ def test_strip_shadowing_flags_boolean_does_not_consume_next_token(): def test_strip_shadowing_flags_jinja_boolean_preserves_positional(): - out = strip_shadowing_flags( - ["--jinja", "trailing-positional"], strip_template = True - ) + out = strip_shadowing_flags(["--jinja", "trailing-positional"], strip_template = True) assert out == ["trailing-positional"]