phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-09 19:58:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fixing import path

Browse source
This commit is contained in:
Chantal D Gama Rose 2025-02-19 16:31:30 -08:00
parent 685deeab8f
commit 96240b7a49
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/remote/safety/nvidia/config.py
View file

@ -9,7 +9,7 @@ from typing import Any, Dict, Optional
from pydantic import BaseModel, Field, SecretStr, field_validator
from llama_models.schema_utils import json_schema_type
from llama_stack.schema_utils import json_schema_type
class ShieldType(Enum):

2
llama_stack/providers/tests/safety/fixtures.py
View file

@ -108,7 +108,7 @@ def safety_nvidia() -> ProviderFixture:
)
SAFETY_FIXTURES = ["llama_guard", "bedrock", "remote"]
SAFETY_FIXTURES = ["llama_guard", "bedrock", "remote", "nvidia"]
@pytest_asyncio.fixture(scope="session")