Fixed for linters

Author: Юдыцкий Игорь

taskiq_pipelines/middleware.py

@@ -1,5 +1,5 @@
 from logging import getLogger
-from typing import Any, List
+from typing import Any, List, Optional
 
 import pydantic
 from taskiq import TaskiqMessage, TaskiqMiddleware, TaskiqResult
@@ -110,7 +110,11 @@ async def on_error(
             return
         await self.fail_pipeline(steps[-1].task_id, result.error)
 
-    async def fail_pipeline(self, last_task_id: str, abort: AbortPipeline | None = None) -> None:
+    async def fail_pipeline(
+        self,
+        last_task_id: str,
+        abort: Optional[BaseException] = None,
+    ) -> None:
         """
         This function aborts pipeline.
 

tests/test_steps.py

@@ -1,10 +1,9 @@
-import sys
 from typing import List
 
 import pytest
 from taskiq import InMemoryBroker
 
-from taskiq_pipelines import Pipeline, PipelineMiddleware, AbortPipeline
+from taskiq_pipelines import AbortPipeline, Pipeline, PipelineMiddleware
 
 
 @pytest.mark.anyio
@@ -49,7 +48,7 @@ def double(i: int) -> int:
 async def test_abort_pipeline() -> None:
     """Test AbortPipeline."""
     broker = InMemoryBroker().with_middlewares(PipelineMiddleware())
-    text = 'task was aborted'
+    text = "task was aborted"
 
     @broker.task
     def normal_task(i: bool) -> bool: