diff --git a/tests/test_litellm/llms/bedrock/passthrough/test_bedrock_passthrough_transformation.py b/tests/test_litellm/llms/bedrock/passthrough/test_bedrock_passthrough_transformation.py index d564bff0032..b7833c86c43 100644 --- a/tests/test_litellm/llms/bedrock/passthrough/test_bedrock_passthrough_transformation.py +++ b/tests/test_litellm/llms/bedrock/passthrough/test_bedrock_passthrough_transformation.py @@ -523,7 +523,9 @@ def test_handle_logging_collected_chunks_inference_profile_arn_falls_back_gracef # Intercept the decoder so we don't need real botocore event-stream bytes. # The import is done inside handle_logging_collected_chunks, so patch the source module. - with patch("litellm.llms.bedrock.chat.get_bedrock_event_stream_decoder") as mock_decoder: + with patch( # test-quality-ok: the behavior under test is which invoke_provider this factory receives + "litellm.llms.bedrock.chat.get_bedrock_event_stream_decoder" + ) as mock_decoder: mock_chunk_obj = MagicMock() mock_chunk_obj._chunk_parser.return_value = {} # not a valid GenericStreamingChunk mock_decoder.return_value = mock_chunk_obj