From 493020eed859bd20e49f4ac7ec0b6c3293c7d3fd Mon Sep 17 00:00:00 2001 From: Robert Craigie Date: Wed, 27 Nov 2024 10:32:27 +0000 Subject: [PATCH] chore(internal): minor formatting changes --- tests/lib/test_bedrock.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/lib/test_bedrock.py b/tests/lib/test_bedrock.py index bb29f106..280cb3ae 100644 --- a/tests/lib/test_bedrock.py +++ b/tests/lib/test_bedrock.py @@ -92,6 +92,7 @@ async def test_messages_retries_async(respx_mock: MockRouter) -> None: == "https://bedrock-runtime.us-east-1.amazonaws.com/model/anthropic.claude-3-sonnet-20240229-v1:0/invoke" ) + @pytest.mark.respx() def test_application_inference_profile(respx_mock: MockRouter) -> None: respx_mock.post(re.compile(r"https://bedrock-runtime\.us-east-1\.amazonaws\.com/model/.*/invoke")).mock( @@ -113,7 +114,6 @@ def test_application_inference_profile(respx_mock: MockRouter) -> None: ) calls = cast("list[MockRequestCall]", respx_mock.calls) - assert len(calls) == 2 assert ( @@ -123,4 +123,4 @@ def test_application_inference_profile(respx_mock: MockRouter) -> None: assert ( calls[1].request.url == "https://bedrock-runtime.us-east-1.amazonaws.com/model/arn:aws:bedrock:us-east-1:123456789012:application-inference-profile%2Fjf2sje1c0jnb/invoke" - ) \ No newline at end of file + )