fix: type issues

Author: HelloJowet

tests/client/test_auth_middleware.py

@@ -1,9 +1,8 @@
-from a2a.client.auth.credentials import InMemoryContextCredentialStore
 import json
 
-from collections.abc import Callable
+from collections.abc import Callable, Generator
 from dataclasses import dataclass
-from typing import Any, Generator
+from typing import Any
 
 import httpx
 import pytest
@@ -18,6 +17,7 @@
     ClientFactory,
     InMemoryContextCredentialStore,
 )
+from a2a.client.auth.credentials import InMemoryContextCredentialStore
 from a2a.types import (
     APIKeySecurityScheme,
     AgentCapabilities,

tests/server/tasks/test_task_updater.py

@@ -1,6 +1,3 @@
-from a2a.server.tasks.task_updater import TaskUpdater
-
-
 import asyncio
 import uuid
 
@@ -11,6 +8,7 @@
 from a2a.server.events import EventQueue
 from a2a.server.id_generator import IDGenerator
 from a2a.server.tasks import TaskUpdater
+from a2a.server.tasks.task_updater import TaskUpdater
 from a2a.types import (
     Message,
     Part,

tests/utils/test_telemetry.py

@@ -1,10 +1,8 @@
-from unittest.mock import MagicMock
-
-
 import asyncio
 
 from typing import NoReturn
 from unittest import mock
+from unittest.mock import MagicMock
 
 import pytest