lint

Signed-off-by: Filinto Duran <[email protected]>

Author: filintod

durabletask/__init__.py

@@ -1,12 +1,9 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
+# Public async exports (import directly from durabletask.aio)
+from durabletask.aio import AsyncWorkflowContext, CoroutineOrchestratorRunner  # noqa: F401
+
 """Durable Task SDK for Python"""
 
 PACKAGE_NAME = 'durabletask'
-
-# Public async exports (import directly from durabletask.aio)
-from durabletask.aio import (  # noqa: F401
-    AsyncWorkflowContext,
-    CoroutineOrchestratorRunner,
-)

durabletask/client.py

@@ -240,9 +240,6 @@ def wait_for_orchestration_completion(
                     OrchestrationStatus.FAILED,
                     OrchestrationStatus.TERMINATED,
                 ]:
-                    print(
-                        f'[CLIENT DEBUG] Orchestration {instance_id} already in terminal state: {current_state.runtime_status}'
-                    )
                     return current_state
 
                 # Poll for completion with exponential backoff to handle eventual consistency

durabletask/internal/helpers.py

@@ -218,8 +218,9 @@ def new_complete_orchestration_action(
     )
 
     return pb.OrchestratorAction(
-        id=id, completeOrchestration=completeOrchestrationAction, 
-        router=pb.TaskRouter(sourceAppID=app_id) if app_id else None
+        id=id,
+        completeOrchestration=completeOrchestrationAction,
+        router=pb.TaskRouter(sourceAppID=app_id) if app_id else None,
     )
 
 

durabletask/worker.py

@@ -860,17 +860,7 @@ def resume(self):
                 # This will either return a Task or raise StopIteration if it's done.
                 try:
                     _val = self._previous_task.get_result()
-                    import os as _os
-
-                    if _os.getenv('DAPR_WF_DEBUG') or _os.getenv('DT_DEBUG'):
-                        print(
-                            f'[DT] resume send instance={self._instance_id} type={type(_val)} is_none={_val is None}'
-                        )
-                except Exception as _e:
-                    import os as _os
-
-                    if _os.getenv('DAPR_WF_DEBUG') or _os.getenv('DT_DEBUG'):
-                        print(f'[DT] resume send error instance={self._instance_id} err={_e}')
+                except Exception:
                     raise
                 next_task = self._generator.send(_val)
 
@@ -901,7 +891,9 @@ def set_complete(
         if result is not None:
             result_json = result if is_result_encoded else shared.to_json(result)
         action = ph.new_complete_orchestration_action(
-            self.next_sequence_number(), status, result_json,
+            self.next_sequence_number(),
+            status,
+            result_json,
             app_id=self._app_id,
         )
         self._pending_actions[action.id] = action
@@ -1445,16 +1437,6 @@ def process_event(self, ctx: _RuntimeOrchestrationContext, event: pb.HistoryEven
                 result = None
                 if not ph.is_empty(event.taskCompleted.result):
                     result = shared.from_json(event.taskCompleted.result.value)
-                try:
-                    import os as _os
-
-                    if _os.getenv('DAPR_WF_DEBUG') or _os.getenv('DT_DEBUG'):
-                        print(
-                            f'[DT] taskCompleted decode instance={ctx.instance_id} task_id={task_id} type={type(result)} is_none={result is None}'
-                        )
-                        print(f'[DT] pending_task_present={activity_task is not None}')
-                except Exception:
-                    pass
                 activity_task.complete(result)
                 ctx.resume()
             elif event.HasField('taskFailed'):

examples/async_non_determinism_demo.py

@@ -9,7 +9,6 @@
 import os
 import time
 import warnings
-from datetime import datetime
 
 from durabletask.aio import SandboxMode
 from durabletask.client import TaskHubGrpcClient

tests/aio/compatibility_utils.py

@@ -18,7 +18,6 @@
 from unittest.mock import Mock
 
 from durabletask import task
-from durabletask.aio.compatibility import OrchestrationContextProtocol
 
 
 class CompatibilityChecker:

tests/aio/test_async_orchestrator.py

@@ -353,7 +353,7 @@ def test_async_when_any_ignores_losers_deterministically():
     async def orch(ctx, _):
         a = ctx.activity('a', input=1)
         b = ctx.activity('b', input=2)
-        _winner = await ctx.when_any([a, b])
+        await ctx.when_any([a, b])
         return 'done'
 
     def a(_, x):

tests/aio/test_asyncio_compat_enhanced.py

@@ -4,8 +4,7 @@
 
 import asyncio
 import os
-import time
-from datetime import datetime, timedelta
+from datetime import datetime
 from unittest.mock import Mock, patch
 
 import pytest
@@ -114,7 +113,7 @@ def test_activity_logging(self):
         with patch.dict(os.environ, {'DAPR_WF_DEBUG': 'true'}):
             ctx = AsyncWorkflowContext(self.mock_base_ctx)
 
-            awaitable = ctx.activity('test_activity', input='test')
+            ctx.activity('test_activity', input='test')
 
             assert len(ctx._operation_history) == 1
             op = ctx._operation_history[0]
@@ -126,7 +125,7 @@ def test_sleep_logging(self):
         with patch.dict(os.environ, {'DAPR_WF_DEBUG': 'true'}):
             ctx = AsyncWorkflowContext(self.mock_base_ctx)
 
-            awaitable = ctx.sleep(5.0)
+            ctx.sleep(5.0)
 
             assert len(ctx._operation_history) == 1
             op = ctx._operation_history[0]

tests/aio/test_context.py

@@ -5,7 +5,6 @@
 Tests for AsyncWorkflowContext in durabletask.aio.
 """
 
-import inspect
 import random
 import uuid
 from datetime import datetime, timedelta

tests/aio/test_context_compatibility.py

@@ -12,16 +12,13 @@
 
 import inspect
 from datetime import datetime, timedelta
-from typing import Any, Optional, Union
 from unittest.mock import Mock
 
 import pytest
 
 from durabletask import task
 from durabletask.aio import AsyncWorkflowContext
 
-from .compatibility_utils import CompatibilityChecker
-
 
 class TestAsyncWorkflowContextCompatibility:
     """Test suite to validate AsyncWorkflowContext compatibility with OrchestrationContext."""