diff --git a/litellm/router.py b/litellm/router.py
index 20c9018abee..e85ca1bd7a8 100644
--- a/litellm/router.py
+++ b/litellm/router.py
@@ -6175,8 +6175,7 @@ class Router:
                         kwargs=new_kwargs,
                         function_name="aretrieve_batch",
                     )
-                    # A batch is retrieved by id, so only the deployment that answered knows the
-                    # group, and batch token usage is logged on this retrieve call.
+                    # Batch token usage is logged on this retrieve call, not on create.
                     model_group: Final = requested_model_group or model_name["model_name"]
                     new_kwargs[metadata_variable_name].setdefault("model_group", model_group)
                     new_kwargs.pop("custom_llm_provider", None)
diff --git a/tests/test_litellm/test_router.py b/tests/test_litellm/test_router.py
index 258d1c973d0..dc51339cf55 100644
--- a/tests/test_litellm/test_router.py
+++ b/tests/test_litellm/test_router.py
@@ -1009,8 +1009,6 @@ _BATCH_OUTPUT_JSONL = "\n".join(
 
 
 class _BatchPayloadCollector(CustomLogger):
-    """Captures the StandardLoggingPayload the spend log row is built from."""
-
     def __init__(self):
         super().__init__()
         self.payloads = []
@@ -1043,7 +1041,6 @@ def _batch_model_group_router():
 
 
 def _mock_batch_provider(respx_mock):
-    """The completed batch, plus the output file the usage accounting reads."""
     respx_mock.get(f"{_BATCH_API_BASE}/batches/{_BATCH_ID}").mock(
         return_value=httpx.Response(200, json=_BATCH_COMPLETED)
     )