From d9ff4e592218d3b51f82c8427a9b1b1192db8668 Mon Sep 17 00:00:00 2001 From: yaythomas Date: Wed, 4 Feb 2026 08:51:49 -0800 Subject: [PATCH] fix: Import concrete BatchResult Previously, context.py imported BatchResult from the types module, which only exposed the Protocol interface with the get_results() method. This prevented client-side type checkers and IDEs from discovering the full set of methods available on BatchResult objects returned by map() and parallel() operations. Changed to import the concrete BatchResult class from concurrency.models, which exposes all available methods including: - throw_if_error() - succeeded(), failed(), started() - has_failure, status properties - success_count, failure_count, started_count properties - get_errors() This improves developer experience by providing accurate type hints and autocomplete for all BatchResult methods without introducing circular dependencies. --- src/aws_durable_execution_sdk_python/context.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/aws_durable_execution_sdk_python/context.py b/src/aws_durable_execution_sdk_python/context.py index 46d48a1..a1a49a5 100644 --- a/src/aws_durable_execution_sdk_python/context.py +++ b/src/aws_durable_execution_sdk_python/context.py @@ -44,21 +44,21 @@ ) from aws_durable_execution_sdk_python.state import ExecutionState # noqa: TCH001 from aws_durable_execution_sdk_python.threading import OrderedCounter +from aws_durable_execution_sdk_python.types import Callback as CallbackProtocol +from aws_durable_execution_sdk_python.types import ( + DurableContext as DurableContextProtocol, +) from aws_durable_execution_sdk_python.types import ( - BatchResult, LoggerInterface, StepContext, WaitForCallbackContext, WaitForConditionCheckContext, ) -from aws_durable_execution_sdk_python.types import Callback as CallbackProtocol -from aws_durable_execution_sdk_python.types import ( - DurableContext as DurableContextProtocol, -) if TYPE_CHECKING: from collections.abc import Callable, Sequence + from aws_durable_execution_sdk_python.concurrency.models import BatchResult from aws_durable_execution_sdk_python.state import CheckpointedResult from aws_durable_execution_sdk_python.types import LambdaContext from aws_durable_execution_sdk_python.waits import WaitForConditionConfig