diff --git a/studio/install_llama_prebuilt.py b/studio/install_llama_prebuilt.py index ee86a985b1..0345bf6622 100644 --- a/studio/install_llama_prebuilt.py +++ b/studio/install_llama_prebuilt.py @@ -1673,7 +1673,9 @@ def resolve_lemonade_rocm_choice( def _lemonade_release_plans( - llama_tag: str, host: HostInfo, published_release_tag: str = "" + llama_tag: str, + host: HostInfo, + published_release_tag: str = "", ) -> tuple[str, list[InstallReleasePlan]]: """Single-attempt plan for a data-center AMD GPU, sourced from lemonade. release_tag is lemonade's own counter so updates compare against lemonade; diff --git a/tests/studio/install/test_rocm_support.py b/tests/studio/install/test_rocm_support.py index 338cbc1cca..8a6c64052e 100644 --- a/tests/studio/install/test_rocm_support.py +++ b/tests/studio/install/test_rocm_support.py @@ -333,7 +333,11 @@ class TestResolveUpstreamAssetChoice: # TEST: install_llama_prebuilt.py -- data-center ROCm (lemonade) routing -def _lemonade_release(tag = "b1300", published_at = "2026-01-01T00:00:00Z", gfxs = None): +def _lemonade_release( + tag = "b1300", + published_at = "2026-01-01T00:00:00Z", + gfxs = None, +): base = f"https://github.com/lemonade-sdk/llamacpp-rocm/releases/download/{tag}" arches = gfxs if gfxs is not None else ["gfx908", "gfx90a"] names = [f"llama-{tag}-ubuntu-rocm-{g}-x64.zip" for g in arches] @@ -447,6 +451,7 @@ class TestDataCenterLemonadeRouting: if "lemonade-sdk" in url: return [_lemonade_release(tag = "b1301")] # would win if pin ignored return {"tag_name": "b9637"} + mock_fetch.side_effect = _router host = rocm_host(rocm_gfx_target = "gfx908") _tag, plans = prebuilt_mod.resolve_simple_install_release_plans(