From c1935885c3caee2b27e5bf1757086acb9b3571b8 Mon Sep 17 00:00:00 2001 From: mateo-berri <277851410+mateo-berri@users.noreply.github.com> Date: Fri, 6 Nov 2026 11:40:28 -0800 Subject: [PATCH] test(nvidia_nim): freeze the discovery cache clock --- .../unit/proxy/credential_endpoints/test_nim_model_discovery.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/proxy/credential_endpoints/test_nim_model_discovery.py b/tests/unit/proxy/credential_endpoints/test_nim_model_discovery.py index a5a798663df..107783a60e2 100644 --- a/tests/unit/proxy/credential_endpoints/test_nim_model_discovery.py +++ b/tests/unit/proxy/credential_endpoints/test_nim_model_discovery.py @@ -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