fix test

deepset-ai · Feb 20, 2024 · 8cf174a · 8cf174a
1 parent 818e7bf
commit 8cf174a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integrations/amazon_bedrock/tests/test_amazon_chat_bedrock.py b/integrations/amazon_bedrock/tests/test_amazon_chat_bedrock.py
@@ -58,7 +58,7 @@ def test_to_dict(mock_auto_tokenizer, mock_boto3_session, set_env_variables):
             "model": "anthropic.claude-v2",
             "generation_kwargs": {"temperature": 0.7},
             "stop_words": [],
-            "streaming_callback": print_streaming_chunk,
+            "streaming_callback": 'haystack.components.generators.utils.print_streaming_chunk',
         },
     }