diff --git a/tests/unit/images/test_image_edit_utils.py b/tests/unit/images/test_image_edit_utils.py index 109c4b89101..2f240b0fed4 100644 --- a/tests/unit/images/test_image_edit_utils.py +++ b/tests/unit/images/test_image_edit_utils.py @@ -536,15 +536,6 @@ async def test_azure_image_edit_litellm_sdk(): # Validate the response meets expected schema ImageResponse.model_validate(result) - if isinstance(result, ImageResponse) and result.data: - image_base64 = result.data[0].b64_json - if image_base64: - image_bytes = base64.b64decode(image_base64) - - # Save the image to a file - with open("test_image_edit.png", "wb") as f: - f.write(image_bytes) - @pytest.mark.asyncio async def test_openai_image_edit_cost_tracking(): """Test OpenAI image edit cost tracking with custom logger""" @@ -607,15 +598,6 @@ async def test_openai_image_edit_cost_tracking(): # Validate the response meets expected schema ImageResponse.model_validate(result) - if isinstance(result, ImageResponse) and result.data: - image_base64 = result.data[0].b64_json - if image_base64: - image_bytes = base64.b64decode(image_base64) - - # Save the image to a file - with open("test_image_edit.png", "wb") as f: - f.write(image_bytes) - await asyncio.sleep(5) print( "standard logging payload",