chore: Formatting

Author: holtskinner

src/a2a/server/events/event_queue.py

@@ -29,9 +29,8 @@ class EventQueue:
     to create child queues that receive the same events.
     """
 
-    def __init__(self, max_queue_size=DEFAULT_MAX_QUEUE_SIZE) -> None:
+    def __init__(self, max_queue_size: int = DEFAULT_MAX_QUEUE_SIZE) -> None:
         """Initializes the EventQueue."""
-
         # Make sure the `asyncio.Queue` is bounded.
         # If it's unbounded (maxsize=0), then `queue.put()` never needs to wait,
         # and so the streaming won't work correctly.

src/a2a/server/tasks/task_updater.py

@@ -109,7 +109,9 @@ async def failed(self, message: Message | None = None):
 
     async def reject(self, message: Message | None = None):
         """Marks the task as rejected and publishes a final status update."""
-        await self.update_status(TaskState.rejected, message=message, final=True)
+        await self.update_status(
+            TaskState.rejected, message=message, final=True
+        )
 
     async def submit(self, message: Message | None = None):
         """Marks the task as submitted and publishes a status update."""

tests/server/tasks/test_task_updater.py

@@ -146,9 +146,7 @@ async def test_complete_without_message(task_updater, event_queue):
 
 
 @pytest.mark.asyncio
-async def test_complete_with_message(
-    task_updater, event_queue, sample_message
-):
+async def test_complete_with_message(task_updater, event_queue, sample_message):
     """Test marking a task as completed with a message."""
     await task_updater.complete(message=sample_message)
 
@@ -176,9 +174,7 @@ async def test_submit_without_message(task_updater, event_queue):
 
 
 @pytest.mark.asyncio
-async def test_submit_with_message(
-    task_updater, event_queue, sample_message
-):
+async def test_submit_with_message(task_updater, event_queue, sample_message):
     """Test marking a task as submitted with a message."""
     await task_updater.submit(message=sample_message)