chore: remove unused invoke timeout

Author: bchampp

src/aws_durable_execution_sdk_python/config.py

@@ -384,13 +384,9 @@ class InvokeConfig(Generic[P, R]):
     Configuration for invoke operations.
 
     This class configures how function invocations are executed, including
-    timeout behavior, serialization, and tenant isolation.
+    serialization and tenant isolation.
 
     Args:
-        timeout: Maximum duration to wait for the invoked function to complete.
-            Default is no timeout. Use this to prevent long-running invocations
-            from blocking execution indefinitely.
-
         serdes_payload: Custom serialization/deserialization for the payload
             sent to the invoked function. Defaults to DEFAULT_JSON_SERDES when
             not set.
@@ -404,16 +400,10 @@ class InvokeConfig(Generic[P, R]):
     """
 
     # retry_strategy: Callable[[Exception, int], RetryDecision] | None = None
-    timeout: Duration = field(default_factory=Duration)
     serdes_payload: SerDes[P] | None = None
     serdes_result: SerDes[R] | None = None
     tenant_id: str | None = None
 
-    @property
-    def timeout_seconds(self) -> int:
-        """Get timeout in seconds."""
-        return self.timeout.to_seconds()
-
 
 @dataclass(frozen=True)
 class CallbackConfig:

src/aws_durable_execution_sdk_python/operation/invoke.py

@@ -166,7 +166,7 @@ def execute(self, _checkpointed_result: CheckpointedResult) -> R:
             ExecutionError: If suspend doesn't raise (should never happen)
         """
         msg: str = f"Invoke {self.operation_identifier.operation_id} started, suspending for completion"
-        suspend_with_optional_resume_delay(msg, self.config.timeout_seconds)
+        suspend_with_optional_resume_delay(msg, None)
         # This line should never be reached since suspend_with_optional_resume_delay always raises
         error_msg: str = "suspend_with_optional_resume_delay should have raised an exception, but did not."
         raise ExecutionError(error_msg) from None

src/aws_durable_execution_sdk_python/serdes.py

@@ -316,7 +316,11 @@ def encode(self, obj: Any) -> EncodedValue:
     def decode(self, tag: TypeTag, value: Any) -> Any:
         match tag:
             case (
-                TypeTag.NONE | TypeTag.STR | TypeTag.BOOL | TypeTag.INT | TypeTag.FLOAT
+                TypeTag.NONE
+                | TypeTag.STR
+                | TypeTag.BOOL
+                | TypeTag.INT
+                | TypeTag.FLOAT
             ):
                 return self.primitive_codec.decode(tag, value)
             case TypeTag.BYTES:

tests/concurrency_test.py

@@ -2833,9 +2833,9 @@ def task_func(ctx, item, idx, items):
     # With tolerated_failure_count=1, executor stops when failure_count > 1 (at 2 failures)
     # Executor terminates early rather than executing all 100 tasks
     assert executed_count["value"] < 100
-    assert result.completion_reason == CompletionReason.FAILURE_TOLERANCE_EXCEEDED, (
-        executed_count
-    )
+    assert (
+        result.completion_reason == CompletionReason.FAILURE_TOLERANCE_EXCEEDED
+    ), executed_count
     assert sum(1 for item in result.all if item.status == BatchItemStatus.FAILED) == 2
     assert (
         sum(1 for item in result.all if item.status == BatchItemStatus.SUCCEEDED) < 98
@@ -2965,9 +2965,9 @@ def slow_branch():
 
     # Slow branch may or may not have started (depends on thread scheduling)
     # but it definitely should not have completed
-    assert operation_tracker.slow_completed.call_count == 0, (
-        "Executor should return before slow branch completes"
-    )
+    assert (
+        operation_tracker.slow_completed.call_count == 0
+    ), "Executor should return before slow branch completes"
 
     # Result should show MIN_SUCCESSFUL_REACHED
     assert result.completion_reason == CompletionReason.MIN_SUCCESSFUL_REACHED
@@ -3019,9 +3019,9 @@ def slow_func():
     result = executor.execute(execution_state, executor_context)
 
     # Executor should have returned before slow branch completed
-    assert not slow_branch_mock.completed.called, (
-        "Executor should return before slow branch completes"
-    )
+    assert (
+        not slow_branch_mock.completed.called
+    ), "Executor should return before slow branch completes"
 
     # Result should show MIN_SUCCESSFUL_REACHED
     assert result.completion_reason == CompletionReason.MIN_SUCCESSFUL_REACHED

tests/config_test.py

@@ -282,7 +282,6 @@ def test_invoke_config_defaults():
     """Test InvokeConfig defaults."""
     config = InvokeConfig()
     assert config.tenant_id is None
-    assert config.timeout_seconds == 0
 
 
 def test_invoke_config_with_tenant_id():

tests/context_test.py

@@ -616,7 +616,7 @@ def test_invoke_with_name_and_config(mock_executor_class):
     mock_state.durable_execution_arn = (
         "arn:aws:durable:us-east-1:123456789012:execution/test"
     )
-    config = InvokeConfig[str, str](timeout=Duration.from_seconds(30))
+    config = InvokeConfig[str, str]()
 
     context = create_test_context(state=mock_state)
     [context._create_step_id() for _ in range(5)]  # Set counter to 5 # noqa: SLF001
@@ -756,7 +756,6 @@ def test_invoke_with_custom_serdes(mock_executor_class):
     config = InvokeConfig[dict, dict](
         serdes_payload=payload_serdes,
         serdes_result=result_serdes,
-        timeout=Duration.from_minutes(1),
     )
 
     context = create_test_context(state=mock_state)

tests/operation/invoke_test.py

@@ -219,9 +219,9 @@ def test_invoke_handler_already_started_with_timeout(status):
     mock_result = CheckpointedResult.create_from_operation(operation)
     mock_state.get_checkpoint_result.return_value = mock_result
 
-    config = InvokeConfig[str, str](timeout=Duration.from_seconds(30))
+    config = InvokeConfig[str, str]()
 
-    with pytest.raises(TimedSuspendExecution):
+    with pytest.raises(SuspendExecution):
         invoke_handler(
             function_name="test_function",
             payload="test_input",
@@ -246,7 +246,7 @@ def test_invoke_handler_new_operation():
     started = CheckpointedResult.create_from_operation(started_op)
     mock_state.get_checkpoint_result.side_effect = [not_found, started]
 
-    config = InvokeConfig[str, str](timeout=Duration.from_minutes(1))
+    config = InvokeConfig[str, str]()
 
     with pytest.raises(
         SuspendExecution, match="Invoke invoke8 started, suspending for completion"
@@ -285,9 +285,9 @@ def test_invoke_handler_new_operation_with_timeout():
     started = CheckpointedResult.create_from_operation(started_op)
     mock_state.get_checkpoint_result.side_effect = [not_found, started]
 
-    config = InvokeConfig[str, str](timeout=Duration.from_seconds(30))
+    config = InvokeConfig[str, str]()
 
-    with pytest.raises(TimedSuspendExecution):
+    with pytest.raises(SuspendExecution):
         invoke_handler(
             function_name="test_function",
             payload="test_input",
@@ -311,7 +311,7 @@ def test_invoke_handler_new_operation_no_timeout():
     started = CheckpointedResult.create_from_operation(started_op)
     mock_state.get_checkpoint_result.side_effect = [not_found, started]
 
-    config = InvokeConfig[str, str](timeout=Duration.from_seconds(0))
+    config = InvokeConfig[str, str]()
 
     with pytest.raises(SuspendExecution):
         invoke_handler(
@@ -1026,7 +1026,7 @@ def test_invoke_immediate_response_with_timeout_immediate_success():
     succeeded = CheckpointedResult.create_from_operation(succeeded_op)
     mock_state.get_checkpoint_result.side_effect = [not_found, succeeded]
 
-    config = InvokeConfig[str, str](timeout=Duration.from_seconds(30))
+    config = InvokeConfig[str, str]()
 
     result = invoke_handler(
         function_name="test_function",
@@ -1061,10 +1061,10 @@ def test_invoke_immediate_response_with_timeout_no_immediate_response():
     started = CheckpointedResult.create_from_operation(started_op)
     mock_state.get_checkpoint_result.side_effect = [not_found, started]
 
-    config = InvokeConfig[str, str](timeout=Duration.from_seconds(30))
+    config = InvokeConfig[str, str]()
 
-    # Verify operation suspends with timeout
-    with pytest.raises(TimedSuspendExecution):
+    # Verify operation suspends
+    with pytest.raises(SuspendExecution):
         invoke_handler(
             function_name="test_function",
             payload="test_input",

tests/operation/wait_for_condition_test.py

@@ -662,9 +662,9 @@ def wait_strategy(state, attempt):
         context_logger=mock_logger,
     )
 
-    assert captured_attempts[-1] == 2, (
-        "After first retry (checkpoint.attempt=1), current attempt should be 2"
-    )
+    assert (
+        captured_attempts[-1] == 2
+    ), "After first retry (checkpoint.attempt=1), current attempt should be 2"
 
     # Test 3: After second retry (checkpoint has attempt=2)
     operation = Operation(
@@ -684,9 +684,9 @@ def wait_strategy(state, attempt):
         context_logger=mock_logger,
     )
 
-    assert captured_attempts[-1] == 3, (
-        "After second retry (checkpoint.attempt=2), current attempt should be 3"
-    )
+    assert (
+        captured_attempts[-1] == 3
+    ), "After second retry (checkpoint.attempt=2), current attempt should be 3"
 
     # Test 4: After third retry (checkpoint has attempt=3)
     operation = Operation(
@@ -706,9 +706,9 @@ def wait_strategy(state, attempt):
         context_logger=mock_logger,
     )
 
-    assert captured_attempts[-1] == 4, (
-        "After third retry (checkpoint.attempt=3), current attempt should be 4"
-    )
+    assert (
+        captured_attempts[-1] == 4
+    ), "After third retry (checkpoint.attempt=3), current attempt should be 4"
 
     # Verify the complete sequence is monotonically increasing
     assert captured_attempts == [

tests/state_test.py

@@ -608,9 +608,9 @@ def test_checkpointed_result_is_timed_out_false_for_other_statuses():
             status=status,
         )
         result = CheckpointedResult.create_from_operation(operation)
-        assert result.is_timed_out() is False, (
-            f"is_timed_out should be False for status {status}"
-        )
+        assert (
+            result.is_timed_out() is False
+        ), f"is_timed_out should be False for status {status}"
 
 
 def test_fetch_paginated_operations_with_marker():
@@ -2535,9 +2535,9 @@ def test_create_checkpoint_is_sync_false_returns_immediately():
     elapsed_time = time.time() - start_time
 
     # Verify it returns immediately (should be < 10ms, we allow 50ms for safety)
-    assert elapsed_time < 0.05, (
-        f"Async checkpoint took {elapsed_time:.3f}s, expected < 0.05s"
-    )
+    assert (
+        elapsed_time < 0.05
+    ), f"Async checkpoint took {elapsed_time:.3f}s, expected < 0.05s"
 
     # Verify operation was enqueued
     assert state._checkpoint_queue.qsize() == 1
@@ -3115,9 +3115,9 @@ def background_processor():
     # Verify all calls blocked for at least the delay time
     for i in range(num_callers):
         elapsed = end_times[i] - start_times[i]
-        assert elapsed >= 0.15, (
-            f"Caller {i} expected blocking for at least 0.15s, got {elapsed}s"
-        )
+        assert (
+            elapsed >= 0.15
+        ), f"Caller {i} expected blocking for at least 0.15s, got {elapsed}s"
 
 
 def test_create_checkpoint_sync_with_empty_checkpoint():