Fix pylint

deepset-ai · Mar 7, 2024 · 21f17ea · 21f17ea
1 parent d56f472
commit 21f17ea
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/integrations/amazon_bedrock/tests/test_chat_generator.py b/integrations/amazon_bedrock/tests/test_chat_generator.py
@@ -229,11 +229,9 @@ def test_get_responses(self) -> None:
 
         assert response_message == [ChatMessage.from_assistant(expected_response)]
 
-    @pytest.mark.parametrize("model_name", [
-        "anthropic.claude-3-sonnet-20240229-v1:0",
-        "anthropic.claude-v2:1",
-        "meta.llama2-13b-chat-v1"
-    ])
+    @pytest.mark.parametrize(
+        "model_name", ["anthropic.claude-3-sonnet-20240229-v1:0", "anthropic.claude-v2:1", "meta.llama2-13b-chat-v1"]
+    )
     @pytest.mark.integration
     def test_default_inference_params(self, model_name):
         messages = [
@@ -249,4 +247,4 @@ def test_default_inference_params(self, model_name):
         assert isinstance(response["replies"][0], ChatMessage)
         assert response["replies"][0].content
         assert ChatMessage.is_from(response["replies"][0], ChatRole.ASSISTANT)
-        assert "paris" in response["replies"][0].content.lower()
+        assert "paris" in response["replies"][0].content.lower()