From 40439509cae6ca3781c7e3a26f4085d1f8df14ab Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Mon, 30 Dec 2024 13:09:38 -0800 Subject: [PATCH] test fixes --- llama_stack/providers/tests/tools/conftest.py | 2 +- llama_stack/providers/tests/tools/test_tools.py | 2 -- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/llama_stack/providers/tests/tools/conftest.py b/llama_stack/providers/tests/tools/conftest.py index 11aad5ab6..6de90dc48 100644 --- a/llama_stack/providers/tests/tools/conftest.py +++ b/llama_stack/providers/tests/tools/conftest.py @@ -10,7 +10,7 @@ from ..conftest import get_provider_fixture_overrides from ..inference.fixtures import INFERENCE_FIXTURES from ..memory.fixtures import MEMORY_FIXTURES from ..safety.fixtures import SAFETY_FIXTURES -from .fixtures import TOOL_RUNTIME_FIXTURES +from .fixtures import TOOL_RUNTIME_FIXTURES, tools_stack # noqa: F401 DEFAULT_PROVIDER_COMBINATIONS = [ pytest.param( diff --git a/llama_stack/providers/tests/tools/test_tools.py b/llama_stack/providers/tests/tools/test_tools.py index 7e4947cb7..f33b4a61d 100644 --- a/llama_stack/providers/tests/tools/test_tools.py +++ b/llama_stack/providers/tests/tools/test_tools.py @@ -14,8 +14,6 @@ from llama_stack.apis.memory_banks import VectorMemoryBankParams from llama_stack.apis.tools import ToolInvocationResult from llama_stack.providers.datatypes import Api -from .fixtures import tools_stack as _tools_stack # noqa: F401, F811 - @pytest.fixture def sample_search_query():