From 0f671ea6b0d08a1f4ddf3b8c2b41357a0dc1ade7 Mon Sep 17 00:00:00 2001 From: Ned Twigg Date: Mon, 16 Dec 2024 10:06:19 -0800 Subject: [PATCH] Fix pyright errors. --- python/example-pytest-selfie/tests/cache_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/example-pytest-selfie/tests/cache_test.py b/python/example-pytest-selfie/tests/cache_test.py index 0222e003..8f458c45 100644 --- a/python/example-pytest-selfie/tests/cache_test.py +++ b/python/example-pytest-selfie/tests/cache_test.py @@ -16,7 +16,7 @@ def openai(): def test_gen_ai(): # Fetch the chat response with caching - chat = cache_selfie_json( + chat: dict = cache_selfie_json( lambda: openai() .chat.completions.create( model="gpt-4o", @@ -63,7 +63,7 @@ def test_gen_ai(): } }""") # raise ValueError(f"KEYS={chat.keys()} TYPE={type(chat)}") - image_url = cache_selfie_json( + image_url: dict = cache_selfie_json( lambda: openai() .images.generate( model="dall-e-3", prompt=chat["choices"][0]["message"]["content"]