Skip to content

Commit 8e4055e

Browse files
committed
refactor: Import concrete BatchResult instead of protocol
Change __init__.py to import BatchResult from concurrency.models (concrete implementation) rather than from types (protocol definition). This provides users with the actual class implementation for better IDE support and type checking. The protocol remains in types.py for internal type checking, while the public API exports the concrete dataclass implementation.
1 parent 06de905 commit 8e4055e

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

src/aws_durable_execution_sdk_python/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22

33
# Main context - used in every durable function
44
# Helper decorators - commonly used for step functions
5+
# Concurrency
6+
from aws_durable_execution_sdk_python.concurrency.models import BatchResult
57
from aws_durable_execution_sdk_python.context import (
68
DurableContext,
79
durable_step,
@@ -20,7 +22,7 @@
2022
from aws_durable_execution_sdk_python.execution import durable_execution
2123

2224
# Essential context types - passed to user functions
23-
from aws_durable_execution_sdk_python.types import BatchResult, StepContext
25+
from aws_durable_execution_sdk_python.types import StepContext
2426

2527
__all__ = [
2628
"BatchResult",

0 commit comments

Comments
 (0)