From 9492fa6a7a21c4043a866d7b4147aff4561dc8e3 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 21 May 2025 20:52:38 -0700 Subject: [PATCH] fix: test --- tests/image_gen_tests/base_image_generation_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/image_gen_tests/base_image_generation_test.py b/tests/image_gen_tests/base_image_generation_test.py index df64d353c71..7ffb2b59dea 100644 --- a/tests/image_gen_tests/base_image_generation_test.py +++ b/tests/image_gen_tests/base_image_generation_test.py @@ -71,7 +71,7 @@ class BaseImageGenTest(ABC): from openai.types.images_response import ImagesResponse - ImagesResponse.model_validate(response.model_dump()) + ImagesResponse.model_validate(dict(response)) for d in response.data: assert isinstance(d, Image)