Address comments

Author: FullyTyped

…execution_sdk_python/concurrency/impl.py …ution_sdk_python/concurrency/executor.pysrc/aws_durable_execution_sdk_python/concurrency/impl.py renamed to src/aws_durable_execution_sdk_python/concurrency/executor.py src/aws_durable_execution_sdk_python/concurrency/impl.py renamed to src/aws_durable_execution_sdk_python/concurrency/executor.py

@@ -7,7 +7,7 @@
 from collections.abc import Callable, Sequence
 from typing import TYPE_CHECKING, Generic, TypeVar
 
-from aws_durable_execution_sdk_python.concurrency.impl import ConcurrentExecutor
+from aws_durable_execution_sdk_python.concurrency.executor import ConcurrentExecutor
 from aws_durable_execution_sdk_python.concurrency.models import (
     BatchResult,
     Executable,

src/aws_durable_execution_sdk_python/operation/map.py

@@ -7,7 +7,7 @@
 from collections.abc import Callable, Sequence
 from typing import TYPE_CHECKING, TypeVar
 
-from aws_durable_execution_sdk_python.concurrency.impl import ConcurrentExecutor
+from aws_durable_execution_sdk_python.concurrency.executor import ConcurrentExecutor
 from aws_durable_execution_sdk_python.concurrency.models import Executable
 from aws_durable_execution_sdk_python.config import ParallelConfig
 from aws_durable_execution_sdk_python.lambda_service import OperationSubType

src/aws_durable_execution_sdk_python/operation/parallel.py

@@ -307,11 +307,9 @@ def encode(self, obj: Any) -> EncodedValue:
                 return self.decimal_codec.encode(obj)
             case datetime() | date():
                 return self.datetime_codec.encode(obj)
-            case list() | tuple() | dict():
+            case list() | tuple() | dict() | BatchResult():
                 return self.container_codec.encode(obj)
             case _:
-                if isinstance(obj, BatchResult):
-                    return self.container_codec.encode(obj)
                 msg = f"Unsupported type: {type(obj)}"
                 raise SerDesError(msg)
 

src/aws_durable_execution_sdk_python/serdes.py

@@ -11,7 +11,7 @@
 
 import pytest
 
-from aws_durable_execution_sdk_python.concurrency.impl import (
+from aws_durable_execution_sdk_python.concurrency.executor import (
     ConcurrentExecutor,
     TimerScheduler,
 )
@@ -2519,7 +2519,7 @@ def create_child_context(operation_id):
         executor_context.create_child_context = create_child_context
 
         with patch(
-            "aws_durable_execution_sdk_python.concurrency.impl.child_handler",
+            "aws_durable_execution_sdk_python.concurrency.executor.child_handler",
             patched_child_handler,
         ):
             executor.execute(execution_state, executor_context)

tests/concurrency_test.py

@@ -6,7 +6,7 @@
 
 import pytest
 
-from aws_durable_execution_sdk_python.concurrency.impl import ConcurrentExecutor
+from aws_durable_execution_sdk_python.concurrency.executor import ConcurrentExecutor
 
 # Mock the executor.execute method to return a BatchResult
 from aws_durable_execution_sdk_python.concurrency.models import (