From 4fa01f97022107889ccdcd9f6a442af9be02cae9 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Thu, 1 Oct 2026 18:47:34 -0700
Subject: [PATCH] test(router): type the prefill check's deployment helper

---
 .../pre_call_checks/test_continuation_prefill_check.py          | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/router_utils/pre_call_checks/test_continuation_prefill_check.py b/tests/unit/router_utils/pre_call_checks/test_continuation_prefill_check.py
index 4cbe48eda54..77a3db58336 100644
--- a/tests/unit/router_utils/pre_call_checks/test_continuation_prefill_check.py
+++ b/tests/unit/router_utils/pre_call_checks/test_continuation_prefill_check.py
@@ -25,7 +25,7 @@ def _synthetic_cost_map_entries() -> None:
     litellm.get_model_info.cache_clear()
 
 
-def _deployment(model: str, dep_id: str) -> dict:
+def _deployment(model: str, dep_id: str) -> dict[str, object]:
     return {"litellm_params": {"model": model}, "model_info": {"id": dep_id}}