diff --git a/litellm/cost_calculator.py b/litellm/cost_calculator.py
index 9bff4a0846d..5c97f93ce85 100644
--- a/litellm/cost_calculator.py
+++ b/litellm/cost_calculator.py
@@ -1901,7 +1901,8 @@ def ocr_cost(
         response: Optional[Any] - response object
 
     Returns:
-        Tuple[float, float]: (prompt cost, completion cost) when priced per token, otherwise the page cost in the first element
+        Tuple[float, float]: (prompt cost, completion cost) when priced per token,
+        otherwise the page cost in the first element
     """
     from litellm.llms.base_llm.ocr.transformation import OCRResponse