diff --git a/studio/backend/routes/settings.py b/studio/backend/routes/settings.py index 839f740c3f..0081864db1 100644 --- a/studio/backend/routes/settings.py +++ b/studio/backend/routes/settings.py @@ -487,7 +487,9 @@ def update_openai_auto_switch_override( legacy_id = _legacy_standalone_gguf_key(payload.model_id) if legacy_id and legacy_id != target_id: set_model_override( - legacy_id, llama_extra_args = [], max_seq_length = None, + legacy_id, + llama_extra_args = [], + max_seq_length = None, ) else: # Save under the key a load resolves to, as the removal branch does. diff --git a/studio/backend/tests/test_openai_auto_switch.py b/studio/backend/tests/test_openai_auto_switch.py index 3ed4e9bb27..5585ce62e0 100644 --- a/studio/backend/tests/test_openai_auto_switch.py +++ b/studio/backend/tests/test_openai_auto_switch.py @@ -5027,11 +5027,13 @@ def test_forget_clears_the_filename_derived_key_a_load_still_reads(monkeypatch): _mock_override_store(monkeypatch) settings.set_model_override( - "/models/Qwen3-8B-Q4_K_M.gguf:q4_k_m", max_seq_length = 8192, + "/models/Qwen3-8B-Q4_K_M.gguf:q4_k_m", + max_seq_length = 8192, ) settings_route.update_openai_auto_switch_override( settings_route.ModelOverridePayload( - model_id = "/models/Qwen3-8B-Q4_K_M.gguf", remove = True, + model_id = "/models/Qwen3-8B-Q4_K_M.gguf", + remove = True, ), "tester", ) @@ -5048,13 +5050,12 @@ def test_forget_leaves_another_file_own_derived_key_alone(monkeypatch): settings.set_model_override("/models/Other-Q4_K_M.gguf:q4_k_m", max_seq_length = 4096) settings_route.update_openai_auto_switch_override( settings_route.ModelOverridePayload( - model_id = "/models/Qwen3-8B-Q4_K_M.gguf", remove = True, + model_id = "/models/Qwen3-8B-Q4_K_M.gguf", + remove = True, ), "tester", ) - assert settings.get_model_override("/models/Other-Q4_K_M.gguf:Q4_K_M")[ - "max_seq_length" - ] == 4096 + assert settings.get_model_override("/models/Other-Q4_K_M.gguf:Q4_K_M")["max_seq_length"] == 4096 def test_forget_of_a_repo_quant_key_derives_nothing(monkeypatch): diff --git a/tests/studio/test_model_picker_contracts.py b/tests/studio/test_model_picker_contracts.py index 8056fd78c0..c674dbc01e 100644 --- a/tests/studio/test_model_picker_contracts.py +++ b/tests/studio/test_model_picker_contracts.py @@ -1567,9 +1567,7 @@ def test_cached_repo_settings_key_follows_the_row_not_the_view(): handle: Run from Discover looked up the snapshot path, found nothing, and loaded with default context, GPU and template.""" page = " ".join(_read("features/hub/hub-page.tsx").split()) - assert ( - 'if (kind !== "cache" && resource.source !== "hub_cache") {' in page - ) + assert 'if (kind !== "cache" && resource.source !== "hub_cache") {' in page def test_detail_settings_defers_a_derived_quant_to_a_fresh_status_read():