Skip to content

Commit 70792a7

Browse files
committed
refactor: Import concrete BatchResult
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. closes #284
1 parent 73d1add commit 70792a7

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)