Fix mypy

Author: jsa34

src/pytest_bdd/gherkin_parser.py

@@ -128,7 +128,7 @@ def from_dict(cls, data: dict[str, Any]) -> Self:
             location=Location.from_dict(data["location"]), rows=[Row.from_dict(row) for row in data.get("rows", [])]
         )
 
-    def transpose(self) -> Self:
+    def transpose(self) -> DataTable:
         # Transpose the cells, turning rows into columns
         if not self.rows:
             return self  # Return itself if there are no rows to transpose

src/pytest_bdd/steps.py

@@ -45,6 +45,7 @@ def _(article):
 import pytest
 from typing_extensions import ParamSpec
 
+from .gherkin_parser import DataTable
 from .parser import Step
 from .parsers import StepParser, get_parser
 from .types import GIVEN, THEN, WHEN
@@ -67,6 +68,7 @@ class StepFunctionContext:
     parser: StepParser
     converters: dict[str, Callable[[str], Any]] = field(default_factory=dict)
     target_fixture: str | None = None
+    data_table: DataTable | None = None
 
 
 def get_step_fixture_name(step: Step) -> str: