lint

Signed-off-by: Tim Li <[email protected]>

Author: timl3136

cadence/worker/_decision_task_handler.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Dict, Any
 
 from cadence.api.v1.common_pb2 import Payload
 from cadence.api.v1.service_worker_pb2 import (

tests/cadence/_internal/workflow/test_workflow_engine_integration.py

@@ -3,13 +3,11 @@
 Integration tests for WorkflowEngine.
 """
 
-import asyncio
 import pytest
 from unittest.mock import Mock, AsyncMock, patch
 from cadence.api.v1.service_worker_pb2 import PollForDecisionTaskResponse
 from cadence.api.v1.common_pb2 import Payload, WorkflowExecution, WorkflowType
 from cadence.api.v1.history_pb2 import History, HistoryEvent, WorkflowExecutionStartedEventAttributes
-from cadence.api.v1.decision_pb2 import Decision
 from cadence._internal.workflow.workflow_engine import WorkflowEngine, DecisionResult
 from cadence.workflow import WorkflowInfo
 from cadence.client import Client
@@ -299,7 +297,6 @@ async def test_process_decision_with_query_results(self, workflow_engine, mock_c
 
         # Mock the decision manager to return decisions with query results
         mock_decisions = [Mock()]
-        query_results = {"query1": "result1"}
 
         with patch.object(workflow_engine._decision_manager, 'collect_pending_decisions', return_value=mock_decisions):
             # Process the decision

tests/cadence/worker/test_decision_task_handler_integration.py

@@ -3,20 +3,16 @@
 Integration tests for DecisionTaskHandler and WorkflowEngine.
 """
 
-import asyncio
 import pytest
-from unittest.mock import Mock, AsyncMock, MagicMock, patch
+from unittest.mock import Mock, AsyncMock, patch
 from cadence.api.v1.service_worker_pb2 import (
-    PollForDecisionTaskResponse,
-    RespondDecisionTaskCompletedRequest,
-    RespondDecisionTaskFailedRequest
+    PollForDecisionTaskResponse
 )
 from cadence.api.v1.common_pb2 import Payload, WorkflowExecution, WorkflowType
 from cadence.api.v1.history_pb2 import History, HistoryEvent, WorkflowExecutionStartedEventAttributes
 from cadence.api.v1.decision_pb2 import Decision
 from cadence.worker._decision_task_handler import DecisionTaskHandler
 from cadence.worker._registry import Registry
-from cadence.workflow import WorkflowInfo
 from cadence.client import Client
 
 

tests/cadence/worker/test_decision_worker_integration.py

@@ -10,7 +10,6 @@
 from cadence.api.v1.common_pb2 import Payload, WorkflowExecution, WorkflowType
 from cadence.api.v1.history_pb2 import History, HistoryEvent, WorkflowExecutionStartedEventAttributes
 from cadence.worker._decision import DecisionWorker
-from cadence.worker._decision_task_handler import DecisionTaskHandler
 from cadence.worker._registry import Registry
 from cadence.client import Client