refactor

Author: Andrei Neagu

packages/service-library/src/servicelib/fastapi/long_running_tasks/_client.py

@@ -137,7 +137,7 @@ def _get_url(self, path: str) -> str:
         return f"{url}"
 
     @retry_on_http_errors
-    async def get_task_status(  # type:ignore[override]
+    async def get_task_status(
         self, task_id: TaskId, *, timeout: PositiveFloat | None = None  # noqa: ASYNC109
     ) -> TaskStatus:
         timeout = timeout or self._client_configuration.default_timeout
@@ -156,7 +156,7 @@ async def get_task_status(  # type:ignore[override]
         return TaskStatus.model_validate(result.json())
 
     @retry_on_http_errors
-    async def get_task_result(  # type:ignore[override]
+    async def get_task_result(
         self, task_id: TaskId, *, timeout: PositiveFloat | None = None  # noqa: ASYNC109
     ) -> Any | None:
         timeout = timeout or self._client_configuration.default_timeout
@@ -175,7 +175,7 @@ async def get_task_result(  # type:ignore[override]
         return result.json()
 
     @retry_on_http_errors
-    async def remove_task(  # type:ignore[override]
+    async def remove_task(
         self, task_id: TaskId, *, timeout: PositiveFloat | None = None  # noqa: ASYNC109
     ) -> None:
         timeout = timeout or self._client_configuration.default_timeout

packages/service-library/tests/fastapi/long_running_tasks/test_long_running_tasks_context_manager.py

@@ -24,7 +24,6 @@
     GenericClientError,
     TaskClientTimeoutError,
     TaskExceptionError,
-    TaskNotFoundError,
 )
 from servicelib.long_running_tasks.models import (
     ProgressMessage,
@@ -48,9 +47,7 @@
 async def _assert_task_removed(
     base_client: HttpClient, task_id: TaskId, router_prefix: str
 ) -> None:
-    with pytest.raises(
-        (GenericClientError, TaskNotFoundError), match=f"No task with {task_id} found"
-    ):
+    with pytest.raises(GenericClientError, match=f"No task with {task_id} found"):
         await base_client.get_task_status(task_id)