Autoformat

jackmpcollins · May 23, 2024 · a674426 · a674426
1 parent 882ec62
commit a674426
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/chat_model/test_litellm_chat_model.py b/tests/chat_model/test_litellm_chat_model.py
@@ -55,6 +55,7 @@ def test_litellm_chat_model_metadata(litellm_success_callback_calls):
  callback_call = litellm_success_callback_calls[-1]
  assert callback_call["kwargs"]["litellm_params"]["metadata"] == {"foo": "bar"}
 
+
 @pytest.mark.litellm_openai
 def test_litellm_chat_model_custom_llm_provider(litellm_success_callback_calls):
  """Test that provided custom_llm_provider is passed to the litellm success callback."""
@@ -64,6 +65,7 @@ def test_litellm_chat_model_custom_llm_provider(litellm_success_callback_calls):
  callback_call = litellm_success_callback_calls[-1]
  assert callback_call["kwargs"]["litellm_params"]["custom_llm_provider"] == "custom"
 
+
 @pytest.mark.parametrize(
  ("prompt", "output_types", "expected_output_type"),
  [