From ffd7932860fa0a6a8b9fbc39b7db9c52a21bf371 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 15 Jun 2026 13:03:27 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/install_llama_prebuilt.py | 4 +++- tests/studio/install/test_rocm_support.py | 7 ++++++- 2 files changed, 9 insertions(+), 2 deletions(-) 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(