From 9345f104889056b2ef6646d65375925a0a3bae03 Mon Sep 17 00:00:00 2001 From: Robert Craigie Date: Thu, 20 Jun 2024 18:52:48 +0100 Subject: [PATCH] fix(tests): add explicit type annotation --- tests/lib/test_assistants.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/lib/test_assistants.py b/tests/lib/test_assistants.py index ac92f17ca3..487b9938c7 100644 --- a/tests/lib/test_assistants.py +++ b/tests/lib/test_assistants.py @@ -40,7 +40,7 @@ def assert_signatures_in_sync( @pytest.mark.parametrize("sync", [True, False], ids=["sync", "async"]) def test_create_and_run_poll_method_definition_in_sync(sync: bool, client: OpenAI, async_client: AsyncOpenAI) -> None: - checking_client = client if sync else async_client + checking_client: OpenAI | AsyncOpenAI = client if sync else async_client assert_signatures_in_sync( checking_client.beta.threads.create_and_run, @@ -50,7 +50,7 @@ def test_create_and_run_poll_method_definition_in_sync(sync: bool, client: OpenA @pytest.mark.parametrize("sync", [True, False], ids=["sync", "async"]) def test_create_and_run_stream_method_definition_in_sync(sync: bool, client: OpenAI, async_client: AsyncOpenAI) -> None: - checking_client = client if sync else async_client + checking_client: OpenAI | AsyncOpenAI = client if sync else async_client assert_signatures_in_sync( checking_client.beta.threads.create_and_run,