scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

test(nvidia_nim): freeze the discovery cache clock

Browse source
This commit is contained in:
mateo-berri 2026-11-06 11:40:28 -08:00
parent ad4905e01c
commit c1935885c3
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/unit/proxy/credential_endpoints/test_nim_model_discovery.py
View file

@ -28,7 +28,7 @@ def client(app: FastAPI) -> Iterator[TestClient]:
with (
patch("litellm.proxy.proxy_server.prisma_client", None),
patch.object(litellm, "credential_list", []),
patch("litellm.proxy.credential_endpoints.nim_model_discovery._CACHE", InMemoryCache()),
patch("litellm.proxy.credential_endpoints.nim_model_discovery._CACHE", InMemoryCache(clock=lambda: 0.0)),
):
with TestClient(app) as test_client:
yield test_client