poe lint

Author: yuandrew

temporalio/worker/_replayer.py

@@ -9,7 +9,6 @@
 from dataclasses import dataclass
 from typing import AsyncIterator, Dict, Mapping, Optional, Sequence, Type
 
-from temporalio.bridge.worker import WorkerTaskTypes
 from typing_extensions import TypedDict
 
 import temporalio.api.history.v1
@@ -19,6 +18,7 @@
 import temporalio.converter
 import temporalio.runtime
 import temporalio.workflow
+from temporalio.bridge.worker import WorkerTaskTypes
 
 from ..common import HeaderCodecBehavior
 from ._interceptor import Interceptor

temporalio/worker/_worker.py

@@ -22,14 +22,14 @@
     cast,
 )
 
-from temporalio.bridge.worker import WorkerTaskTypes
 from typing_extensions import TypeAlias, TypedDict
 
 import temporalio.bridge.worker
 import temporalio.client
 import temporalio.common
 import temporalio.runtime
 import temporalio.service
+from temporalio.bridge.worker import WorkerTaskTypes
 from temporalio.common import (
     HeaderCodecBehavior,
     VersioningBehavior,

tests/contrib/openai_agents/test_openai.py

@@ -2615,7 +2615,8 @@ async def test_split_workers(client: Client):
 
     # Workflow worker
     async with new_worker(
-        workflow_client, HelloWorldAgent,
+        workflow_client,
+        HelloWorldAgent,
     ) as worker:
         activity_plugin = openai_agents.OpenAIAgentsPlugin(
             model_params=ModelActivityParameters(

tests/test_plugins.py

@@ -150,7 +150,8 @@ async def test_worker_plugin_basic_config(client: Client) -> None:
         activities=[never_run_activity],
         plugins=[MyWorkerPlugin()],
     )
-    assert worker.config().get("task_queue").startswith("replaced_queue")
+    task_queue = worker.config().get("task_queue")
+    assert task_queue is not None and task_queue.startswith("replaced_queue")
     assert [p.name() for p in worker.config().get("plugins", [])] == [
         MyWorkerPlugin().name()
     ]
@@ -159,8 +160,11 @@ async def test_worker_plugin_basic_config(client: Client) -> None:
     new_config = client.config()
     new_config["plugins"] = [MyCombinedPlugin()]
     client = Client(**new_config)
-    worker = Worker(client, task_queue="queue" + str(uuid.uuid4()), activities=[never_run_activity])
-    assert worker.config().get("task_queue").startswith("combined")
+    worker = Worker(
+        client, task_queue="queue" + str(uuid.uuid4()), activities=[never_run_activity]
+    )
+    task_queue = worker.config().get("task_queue")
+    assert task_queue is not None and task_queue.startswith("combined")
     assert [p.name() for p in worker.config().get("plugins", [])] == [
         MyCombinedPlugin().name()
     ]
@@ -172,7 +176,8 @@ async def test_worker_plugin_basic_config(client: Client) -> None:
         activities=[never_run_activity],
         plugins=[MyWorkerPlugin()],
     )
-    assert worker.config().get("task_queue").startswith("replaced_queue")
+    task_queue = worker.config().get("task_queue")
+    assert task_queue is not None and task_queue.startswith("replaced_queue")
     assert [p.name() for p in worker.config().get("plugins", [])] == [
         MyCombinedPlugin().name(),
         MyWorkerPlugin().name(),
@@ -401,6 +406,10 @@ def configure_worker(self, config: WorkerConfig) -> WorkerConfig:
 async def test_medium_plugin(client: Client) -> None:
     plugin = MediumPlugin()
     worker = Worker(
-        client, task_queue="queue" + str(uuid.uuid4()), plugins=[plugin], workflows=[HelloWorkflow]
+        client,
+        task_queue="queue" + str(uuid.uuid4()),
+        plugins=[plugin],
+        workflows=[HelloWorkflow],
     )
-    assert worker.config().get("task_queue").startswith("override")
+    task_queue = worker.config().get("task_queue")
+    assert task_queue is not None and task_queue.startswith("override")

tests/test_runtime.py

@@ -189,7 +189,9 @@ async def has_log() -> bool:
     # Check record
     record = next((l for l in log_queue_list if "Failing workflow task" in l.message))
     assert record.levelno == logging.WARNING
-    assert record.name == f"{logger.name}-sdk_core::temporalio_sdk_core::worker::workflow"
+    assert (
+        record.name == f"{logger.name}-sdk_core::temporalio_sdk_core::worker::workflow"
+    )
     assert record.temporal_log.fields["run_id"] == handle.result_run_id  # type: ignore