Upgrade mypy to work with pydantic

Author: dandavison

poetry.lock

@@ -22,10 +22,12 @@ temporalio = "^1.10.0"
 [tool.poetry.dev-dependencies]
 black = "^22.3.0"
 isort = "^5.10.1"
-mypy = "^0.981"
+mypy = "^1.4.1"
 pytest = "^7.1.2"
 pytest-asyncio = "^0.18.3"
 frozenlist = "^1.4.0"
+types-pyyaml = "^6.0.12.20241230"
+
 
 # All sample-specific dependencies are in optional groups below, named after the
 # sample they apply to

pyproject.toml

@@ -37,8 +37,10 @@ async def execute_activity(self, input: ExecuteActivityInput) -> Any:
             try:
                 return await super().execute_activity(input)
             except Exception as e:
-                if len(input.args) == 1 and is_dataclass(input.args[0]):
-                    set_context("temporal.activity.input", asdict(input.args[0]))
+                if len(input.args) == 1:
+                    [arg] = input.args
+                    if is_dataclass(arg) and not isinstance(arg, type):
+                        set_context("temporal.activity.input", asdict(arg))
                 set_context("temporal.activity.info", activity.info().__dict__)
                 capture_exception()
                 raise e
@@ -58,8 +60,10 @@ async def execute_workflow(self, input: ExecuteWorkflowInput) -> Any:
             try:
                 return await super().execute_workflow(input)
             except Exception as e:
-                if len(input.args) == 1 and is_dataclass(input.args[0]):
-                    set_context("temporal.workflow.input", asdict(input.args[0]))
+                if len(input.args) == 1:
+                    [arg] = input.args
+                    if is_dataclass(arg) and not isinstance(arg, type):
+                        set_context("temporal.workflow.input", asdict(arg))
                 set_context("temporal.workflow.info", workflow.info().__dict__)
 
                 if not workflow.unsafe.is_replaying():

sentry/interceptor.py

@@ -79,8 +79,8 @@ async def test_safe_message_handlers(client: Client, env: WorkflowEnvironment):
 
         await cluster_manager_handle.signal(ClusterManagerWorkflow.shutdown_cluster)
 
-        result = await cluster_manager_handle.result()
-        assert result.num_currently_assigned_nodes == 0
+        cluster_manager_result = await cluster_manager_handle.result()
+        assert cluster_manager_result.num_currently_assigned_nodes == 0
 
 
 async def test_update_idempotency(client: Client, env: WorkflowEnvironment):