fix: update transformation implementations to match base class

Co-Authored-By: [email protected] <[email protected]>

airbyte_cdk/sources/declarative/transformations/add_fields.py

@@ -5,6 +5,8 @@
 from dataclasses import InitVar, dataclass, field
 from typing import Any, Dict, List, Mapping, Optional, Type, Union
 
+from typing_extensions import override
+
 import dpath
 
 from airbyte_cdk.sources.declarative.interpolation.interpolated_string import InterpolatedString
@@ -119,6 +121,7 @@ def __post_init__(self, parameters: Mapping[str, Any]) -> None:
                     )
                 )
 
+    @override
     def transform(
         self,
         record: Dict[str, Any],

airbyte_cdk/sources/declarative/transformations/dpath_flatten_fields.py

@@ -1,6 +1,8 @@
 from dataclasses import InitVar, dataclass
 from typing import Any, Dict, List, Mapping, Optional, Union
 
+from typing_extensions import override
+
 import dpath
 
 from airbyte_cdk.sources.declarative.interpolation.interpolated_string import InterpolatedString
@@ -33,6 +35,7 @@ def __post_init__(self, parameters: Mapping[str, Any]) -> None:
                     self.field_path[path_index], parameters=parameters
                 )
 
+    @override
     def transform(
         self,
         record: Dict[str, Any],

airbyte_cdk/sources/declarative/transformations/flatten_fields.py

@@ -3,7 +3,9 @@
 #
 
 from dataclasses import dataclass
-from typing import Any, Dict, Optional
+from typing import Any, Dict, Mapping, Optional
+
+from typing_extensions import override
 
 from airbyte_cdk.sources.declarative.transformations import RecordTransformation
 from airbyte_cdk.sources.types import Config, StreamSlice, StreamState
@@ -13,6 +15,7 @@
 class FlattenFields(RecordTransformation):
     flatten_lists: bool = True
 
+    @override
     def transform(
         self,
         record: Dict[str, Any],

airbyte_cdk/sources/declarative/transformations/keys_replace_transformation.py

@@ -5,6 +5,8 @@
 from dataclasses import InitVar, dataclass
 from typing import Any, Dict, Mapping, Optional
 
+from typing_extensions import override
+
 from airbyte_cdk import InterpolatedString
 from airbyte_cdk.sources.declarative.transformations import RecordTransformation
 from airbyte_cdk.sources.types import Config, StreamSlice, StreamState
@@ -32,6 +34,7 @@ def __post_init__(self, parameters: Mapping[str, Any]) -> None:
         self._old = InterpolatedString.create(self.old, parameters=parameters)
         self._new = InterpolatedString.create(self.new, parameters=parameters)
 
+    @override
     def transform(
         self,
         record: Dict[str, Any],

airbyte_cdk/sources/declarative/transformations/keys_to_lower_transformation.py

@@ -3,14 +3,17 @@
 #
 
 from dataclasses import dataclass
-from typing import Any, Dict, Optional
+from typing import Any, Dict, Mapping, Optional
+
+from typing_extensions import override
 
 from airbyte_cdk.sources.declarative.transformations import RecordTransformation
 from airbyte_cdk.sources.types import Config, StreamSlice, StreamState
 
 
 @dataclass
 class KeysToLowerTransformation(RecordTransformation):
+    @override
     def transform(
         self,
         record: Dict[str, Any],

airbyte_cdk/sources/declarative/transformations/keys_to_snake_transformation.py

@@ -4,7 +4,9 @@
 
 import re
 from dataclasses import dataclass
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Mapping, Optional
+
+from typing_extensions import override
 
 import unidecode
 
@@ -18,6 +20,7 @@ class KeysToSnakeCaseTransformation(RecordTransformation):
         r"[A-Z]+[a-z]*|[a-z]+|\d+|(?P<NoToken>[^a-zA-Z\d]+)"
     )
 
+    @override
     def transform(
         self,
         record: Dict[str, Any],

airbyte_cdk/sources/declarative/transformations/remove_fields.py

@@ -5,6 +5,8 @@
 from dataclasses import InitVar, dataclass
 from typing import Any, Dict, List, Mapping, Optional
 
+from typing_extensions import override
+
 import dpath
 import dpath.exceptions
 
@@ -49,6 +51,7 @@ def __post_init__(self, parameters: Mapping[str, Any]) -> None:
             condition=self.condition, parameters=parameters
         )
 
+    @override
     def transform(
         self,
         record: Dict[str, Any],