phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-13 04:22:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added tests for persistence

Browse source
This commit is contained in:
Sarthak Deshpande 2024-10-21 15:56:54 +05:30
parent cae5b0708b
commit 862ff6c653
7 changed files with 157 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/tests/safety/test_safety.py
View file

@ -36,8 +36,8 @@ async def safety_settings():
impls = await resolve_impls_for_test(Api.safety, deps=[Api.inference])
return {
"impl": impls[Api.safety],
"shields_impl": impls[Api.shields],
"impl": impls['impls'][Api.safety],
"shields_impl": impls['impls'][Api.shields],
}