@@ -976,7 +976,11 @@ def test_parallel_handler_serializes_batch_result():
976976 child_checkpoint .is_replay_children .return_value = False
977977
978978 def get_checkpoint (op_id ):
979- return child_checkpoint if op_id .startswith ("child-" ) else parent_checkpoint
979+ return (
980+ child_checkpoint
981+ if op_id .startswith ("child-" )
982+ else parent_checkpoint
983+ )
980984
981985 mock_state = Mock ()
982986 mock_state .durable_execution_arn = "arn:test"
@@ -1030,7 +1034,11 @@ def test_parallel_default_serdes_serializes_batch_result():
10301034 child_checkpoint .is_replay_children .return_value = False
10311035
10321036 def get_checkpoint (op_id ):
1033- return child_checkpoint if op_id .startswith ("child-" ) else parent_checkpoint
1037+ return (
1038+ child_checkpoint
1039+ if op_id .startswith ("child-" )
1040+ else parent_checkpoint
1041+ )
10341042
10351043 mock_state = Mock ()
10361044 mock_state .durable_execution_arn = "arn:test"
@@ -1072,7 +1080,9 @@ def test_parallel_custom_serdes_serializes_batch_result():
10721080 custom_serdes = CustomStrSerDes ()
10731081
10741082 try :
1075- with patch ("aws_durable_execution_sdk_python.serdes.serialize" ) as mock_serialize :
1083+ with patch (
1084+ "aws_durable_execution_sdk_python.serdes.serialize"
1085+ ) as mock_serialize :
10761086 mock_serialize .return_value = '"serialized"'
10771087 importlib .reload (child )
10781088
@@ -1089,7 +1099,11 @@ def test_parallel_custom_serdes_serializes_batch_result():
10891099 child_checkpoint .is_replay_children .return_value = False
10901100
10911101 def get_checkpoint (op_id ):
1092- return child_checkpoint if op_id .startswith ("child-" ) else parent_checkpoint
1102+ return (
1103+ child_checkpoint
1104+ if op_id .startswith ("child-" )
1105+ else parent_checkpoint
1106+ )
10931107
10941108 mock_state = Mock ()
10951109 mock_state .durable_execution_arn = "arn:test"
0 commit comments