Commit 203f094b

Robert Craigie <robert@craigie.dev>
2024-06-21 01:52:48
fix(tests): add explicit type annotation
main
1 parent 717e318
Changed files (1)
tests
lib
test_assistants.py
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,