phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 17:29:01 +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
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:
# 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.
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 (
Checkpoint,
DataConfig,