Fixed flake.

Signed-off-by: Pavel Kirilin <[email protected]>

Author: s3rius

.flake8

@@ -106,6 +106,10 @@ per-file-ignores =
   DAR101,
   ; Found too many arguments
   WPS211,
+  ; Found nested function
+  WPS430,
+  ; Found too short name
+  WPS111,
 
   ; all init files
   __init__.py:

.pre-commit-config.yaml

@@ -41,7 +41,7 @@ repos:
         language: system
         pass_filenames: false
         types: [python]
-        args: [--count, taskiq_pipelines]
+        args: [--count, taskiq_pipelines, tests]
 
       - id: mypy
         name: Validate types with MyPy

taskiq_pipelines/pipeliner.py

@@ -111,7 +111,7 @@ def call_next(
         """
         self.steps.append(
             DumpedStep(
-                step_type=SequentialStep._step_name,
+                step_type=SequentialStep._step_name,  # noqa: WPS437
                 step_data=SequentialStep.from_task(
                     task=task,
                     param_name=param_name,
@@ -167,7 +167,7 @@ def call_after(
         """
         self.steps.append(
             DumpedStep(
-                step_type=SequentialStep._step_name,
+                step_type=SequentialStep._step_name,  # noqa: WPS437
                 step_data=SequentialStep.from_task(
                     task=task,
                     param_name=EMPTY_PARAM_NAME,
@@ -236,7 +236,7 @@ def map(
         """
         self.steps.append(
             DumpedStep(
-                step_type=MapperStep._step_name,
+                step_type=MapperStep._step_name,  # noqa: WPS437
                 step_data=MapperStep.from_task(
                     task=task,
                     param_name=param_name,
@@ -308,7 +308,7 @@ def filter(
         """
         self.steps.append(
             DumpedStep(
-                step_type=FilterStep._step_name,
+                step_type=FilterStep._step_name,  # noqa: WPS437
                 step_data=FilterStep.from_task(
                     task=task,
                     param_name=param_name,

taskiq_pipelines/steps/__init__.py

@@ -10,7 +10,7 @@
 
 
 def parse_step(step_type: str, step_data: str) -> AbstractStep:
-    step_cls = AbstractStep._known_steps.get(step_type)
+    step_cls = AbstractStep._known_steps.get(step_type)  # noqa: WPS437
     if step_cls is None:
         logger.warning(f"Unknown step type: {step_type}")
         raise ValueError("Unknown step type.")

tests/test_steps.py

@@ -8,8 +8,7 @@
 
 @pytest.mark.anyio
 async def test_success() -> None:
-    """Test stub for CI."""
-
+    """Tests that sequential step works as expected."""
     broker = InMemoryBroker().with_middlewares(PipelineMiddleware())
 
     @broker.task
@@ -28,8 +27,7 @@ def double(i: int) -> int:
 
 @pytest.mark.anyio
 async def test_mapping_success() -> None:
-    """Test stub for CI."""
-
+    """Test that map step works as expected."""
     broker = InMemoryBroker().with_middlewares(PipelineMiddleware())
 
     @broker.task