lint

Signed-off-by: Prithvi Kannan <[email protected]>
langchain-ai · Dec 12, 2024 · 59f34f7 · 59f34f7
1 parent a0b84dd
commit 59f34f7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/libs/databricks/tests/integration_tests/test_chat_models.py b/libs/databricks/tests/integration_tests/test_chat_models.py
@@ -47,7 +47,10 @@ def test_chat_databricks_invoke():
     assert response.content == "To learn "
     assert 20 <= response.response_metadata["prompt_tokens"] <= 30
     assert 1 <= response.response_metadata["completion_tokens"] <= 10
-    expected_total = response.response_metadata["prompt_tokens"] + response.response_metadata["completion_tokens"]
+    expected_total = (
+        response.response_metadata["prompt_tokens"]
+        + response.response_metadata["completion_tokens"]
+    )
     assert response.response_metadata["total_tokens"] == expected_total
 
     response = chat.invoke(