Format

Author: donald-pinckney

tests/conftest.py

@@ -36,12 +36,14 @@ def event_loop():
     yield loop
     loop.close()
 
+
 @pytest.fixture(scope="session")
 def plugins():
     # By default, no plugins.
     # Other tests can override this fixture, such as in tests/openai_agents/conftest.py
     return []
 
+
 @pytest_asyncio.fixture(scope="session")
 async def env(request, plugins) -> AsyncGenerator[WorkflowEnvironment, None]:
     env_type = request.config.getoption("--workflow-environment")
@@ -53,12 +55,14 @@ async def env(request, plugins) -> AsyncGenerator[WorkflowEnvironment, None]:
                 "--dynamic-config-value",
                 "system.enableEagerWorkflowStart=true",
             ],
-            plugins=plugins
+            plugins=plugins,
         )
     elif env_type == "time-skipping":
         env = await WorkflowEnvironment.start_time_skipping()
     else:
-        env = WorkflowEnvironment.from_client(await Client.connect(env_type, plugins=plugins))
+        env = WorkflowEnvironment.from_client(
+            await Client.connect(env_type, plugins=plugins)
+        )
     yield env
     await env.shutdown()
 

tests/openai_agents/test_hello_world_workflow.py

@@ -7,7 +7,10 @@
 from temporalio.worker import Worker
 
 from openai_agents.basic.activities.get_weather_activity import get_weather
-from openai_agents.basic.activities.math_activities import multiply_by_two, random_number
+from openai_agents.basic.activities.math_activities import (
+    multiply_by_two,
+    random_number,
+)
 from openai_agents.basic.activities.image_activities import read_image_as_base64
 
 from openai_agents.basic.workflows.hello_world_workflow import HelloWorldAgent
@@ -35,4 +38,3 @@ async def test_execute_workflow(client: Client):
             id=str(uuid.uuid4()),
             task_queue=task_queue_name,
         )
-