From e85481f20729c6a3ab9a1368540dd9e15fd4c85d Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Mon, 4 Aug 2025 11:21:20 -0700 Subject: [PATCH] fix unit test --- tests/unit/distribution/test_build_path.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/distribution/test_build_path.py b/tests/unit/distribution/test_build_path.py index 0d245e818..52a71286b 100644 --- a/tests/unit/distribution/test_build_path.py +++ b/tests/unit/distribution/test_build_path.py @@ -16,8 +16,8 @@ from llama_stack.core.utils.image_types import LlamaStackImageType def test_container_build_passes_path(monkeypatch, tmp_path): called_with = {} - def spy_build_image(build_config, image_name, template_or_config, run_config=None): - called_with["path"] = template_or_config + def spy_build_image(build_config, image_name, distro_or_config, run_config=None): + called_with["path"] = distro_or_config called_with["run_config"] = run_config return 0