diff --git a/tests/api/conftest.py b/tests/api/conftest.py index 8b5e4e7c0..52064fed4 100644 --- a/tests/api/conftest.py +++ b/tests/api/conftest.py @@ -9,14 +9,15 @@ import os from pathlib import Path import pytest -from fixtures.recordable_mock import RecordableMock from llama_stack_client import LlamaStackClient -from report import Report from llama_stack import LlamaStackAsLibraryClient from llama_stack.apis.datatypes import Api from llama_stack.providers.tests.env import get_env_or_fail +from .fixtures.recordable_mock import RecordableMock +from .report import Report + def pytest_configure(config): config.option.tbstyle = "short" diff --git a/tests/api/report.py b/tests/api/report.py index 0151b3d20..762a7afcb 100644 --- a/tests/api/report.py +++ b/tests/api/report.py @@ -13,7 +13,6 @@ from typing import Optional from urllib.parse import urlparse import pytest -from metadata import API_MAPS from pytest import CollectReport from termcolor import cprint @@ -29,6 +28,8 @@ from llama_stack.models.llama.sku_list import ( from llama_stack.providers.datatypes import Api from llama_stack.providers.tests.env import get_env_or_fail +from .metadata import API_MAPS + def featured_models(): models = [