Fix formatting.

Author: denzyll

src/aws_durable_execution_sdk_python/exceptions.py

@@ -146,9 +146,14 @@ def from_exception(cls, exception: Exception) -> Self:
         response = getattr(exception, "response", {})
         response_metadata = response.get("ResponseMetadata")
         error = response.get("Error")
-        error_category = BotoClientError._classify_error_category(error, response_metadata)
+        error_category = BotoClientError._classify_error_category(
+            error, response_metadata
+        )
         return cls(
-            message=str(exception), error_category=error_category, error=error, response_metadata=response_metadata
+            message=str(exception),
+            error_category=error_category,
+            error=error,
+            response_metadata=response_metadata,
         )
 
     @staticmethod
@@ -171,15 +176,19 @@ def _classify_error_category(
         if error_code and error_code in _NON_RETRYABLE_CUSTOMER_ERROR_CODES:
             return DurableApiErrorCategory.EXECUTION
 
-        status_code: int | None = (response_metadata and response_metadata.get("HTTPStatusCode")) or None
+        status_code: int | None = (
+            response_metadata and response_metadata.get("HTTPStatusCode")
+        ) or None
         if (
             status_code
             and BAD_REQUEST_ERROR <= status_code < SERVICE_ERROR
             and status_code != TOO_MANY_REQUESTS_ERROR
             and error
             and not (
                 (error.get("Code") or "") == INVALID_PARAMETER_VALUE_EXCEPTION
-                and (error.get("Message") or "").startswith(INVALID_CHECKPOINT_TOKEN_PREFIX)
+                and (error.get("Message") or "").startswith(
+                    INVALID_CHECKPOINT_TOKEN_PREFIX
+                )
             )
         ):
             return DurableApiErrorCategory.EXECUTION

tests/exceptions_test.py

@@ -123,13 +123,15 @@ def test_checkpoint_error_classification_invalid_param_without_token_execution()
         "KMSNotFoundException",
     ],
 )
-def test_durable_api_error_non_retryable_customer_error_codes(error_cls, error_code: str):
+def test_durable_api_error_non_retryable_customer_error_codes(
+    error_cls, error_code: str
+):
     """Test that non-retryable customer error codes (HTTP 502) are classified as EXECUTION."""
     error_response = {
         "Error": {"Code": error_code, "Message": f"{error_code} error"},
         "ResponseMetadata": {"HTTPStatusCode": 502},
     }
-    client_error = ClientError(error_response, "Invoke")
+    client_error = ClientError(error_response, "Invoke")  # type: ignore[arg-type]
     result = error_cls.from_exception(client_error)
     assert result.error_category == DurableApiErrorCategory.EXECUTION
     assert not result.is_retryable()
@@ -443,13 +445,15 @@ def test_orphaned_child_exception_with_operation_id():
         ("ServiceException", 502, True),
     ],
 )
-def test_boto_client_error_is_retryable(error_code: str, status_code: int, expected_retryable: bool):
+def test_boto_client_error_is_retryable(
+    error_code: str, status_code: int, expected_retryable: bool
+):
     """Test BotoClientError.is_retryable() classification."""
     error_response = {
         "Error": {"Code": error_code, "Message": "test error"},
         "ResponseMetadata": {"HTTPStatusCode": status_code},
     }
-    client_error = ClientError(error_response, "Invoke")
+    client_error = ClientError(error_response, "Invoke")  # type: ignore[arg-type]
     result = BotoClientError.from_exception(client_error)
     assert result.is_retryable() == expected_retryable
 
@@ -480,4 +484,4 @@ def test_durable_api_error_category_backward_compatible_alias():
 def test_invocation_error_is_retryable_default():
     """Test InvocationError.is_retryable() returns True by default."""
     error = InvocationError("some error")
-    assert error.is_retryable()
+    assert error.is_retryable()

tests/execution_test.py

@@ -2667,7 +2667,9 @@ def _make_invocation_input(mock_client, next_marker=""):
     return DurableExecutionInvocationInputWithClient(
         durable_execution_arn="arn:test:execution",
         checkpoint_token="token123",  # noqa: S106
-        initial_execution_state=InitialExecutionState(operations=[operation], next_marker=next_marker),
+        initial_execution_state=InitialExecutionState(
+            operations=[operation], next_marker=next_marker
+        ),
         service_client=mock_client,
     )
 
@@ -2729,7 +2731,9 @@ def test_durable_execution_non_retryable_background_thread_error_returns_failed(
         error={"Code": "KMSDisabledException", "Message": "KMS key disabled"},
         response_metadata={"HTTPStatusCode": 502},
     )
-    mock_client.checkpoint.side_effect = lambda *a, **kw: (_ for _ in ()).throw(non_retryable_error)
+    mock_client.checkpoint.side_effect = lambda *a, **kw: (_ for _ in ()).throw(
+        non_retryable_error
+    )
 
     @durable_execution
     def test_handler(event: Any, context: DurableContext) -> dict:
@@ -2757,7 +2761,7 @@ def test_durable_execution_non_retryable_initial_pagination_error_returns_failed
         message=f"{error_code} error",
         error_category=error_category,
         error={"Code": error_code, "Message": f"{error_code} error"},
-        response_metadata={"HTTPStatusCode": status_code},
+        response_metadata={"HTTPStatusCode": status_code},  # type: ignore[arg-type]
     )
     mock_client.get_execution_state.side_effect = non_retryable_error
 
@@ -2791,4 +2795,4 @@ def test_handler(event: Any, context: DurableContext) -> dict:
         test_handler(
             _make_invocation_input(mock_client, next_marker="next-page-marker"),
             _make_lambda_context(),
-        )
+        )