phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-05 18:22:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix other ImportErrors

Browse source
This commit is contained in:
Connor Hack 2025-01-09 09:06:16 -08:00
parent dc801e827c
commit 5942dffa0f
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/providers/tests/inference/test_embeddings.py
View file

@ -6,7 +6,8 @@
import pytest import pytest
from llama_stack.apis.inference import EmbeddingsResponse, ModelType from llama_stack.apis.inference import EmbeddingsResponse
from llama_stack.apis.models import ModelType
# How to run this test: # How to run this test:
# pytest -v -s llama_stack/providers/tests/inference/test_embeddings.py # pytest -v -s llama_stack/providers/tests/inference/test_embeddings.py

2
llama_stack/providers/tests/post_training/test_post_training.py
View file

@ -5,7 +5,7 @@
# the root directory of this source tree. # the root directory of this source tree.
import pytest import pytest
from llama_stack.apis.common.type_system import JobStatus from llama_stack.apis.common.job_types import JobStatus
from llama_stack.apis.post_training import ( from llama_stack.apis.post_training import (
Checkpoint, Checkpoint,
DataConfig, DataConfig,