mypy and format fix

Author: darynaishchenko

airbyte_cdk/sources/declarative/transformations/dpath_flatten_fields.py

@@ -25,18 +25,18 @@ class DpathFlattenFields(RecordTransformation):
     parameters: InitVar[Mapping[str, Any]]
     delete_origin_value: bool = False
     replace_record: bool = False
-    key_transformation: Union[InterpolatedString, str] = None
+    key_transformation: Union[InterpolatedString, str, None] = None
 
     def __post_init__(self, parameters: Mapping[str, Any]) -> None:
+        self._parameters = parameters
         self._field_path = [
-            InterpolatedString.create(path, parameters=parameters) for path in self.field_path
+            InterpolatedString.create(path, parameters=self._parameters) for path in self.field_path
         ]
         for path_index in range(len(self.field_path)):
             if isinstance(self.field_path[path_index], str):
                 self._field_path[path_index] = InterpolatedString.create(
-                    self.field_path[path_index], parameters=parameters
+                    self.field_path[path_index], parameters=self._parameters
                 )
-        self.parameters = parameters
 
     def transform(
         self,
@@ -53,11 +53,12 @@ def transform(
             extracted = dpath.get(record, path, default=[])
 
         if isinstance(extracted, dict):
-
             if self.key_transformation:
                 updated_extracted = {}
                 for key, value in extracted.items():
-                    updated_key = InterpolatedString.create(self.key_transformation, parameters=self.parameters).eval(key=key, config=self.config)
+                    updated_key = InterpolatedString.create(
+                        self.key_transformation, parameters=self._parameters
+                    ).eval(key=key, config=self.config)
                     updated_extracted[updated_key] = value
                 extracted = updated_extracted
 

unit_tests/sources/declarative/transformations/test_dpath_flatten_fields.py

@@ -9,6 +9,7 @@
 _DO_NOT_REPLACE_WITH_VALUE = False
 _NO_KEY_TRANSFORMATIONS = None
 
+
 @pytest.mark.parametrize(
     [
         "input_record",
@@ -173,7 +174,13 @@
     ],
 )
 def test_dpath_flatten_lists(
-    input_record, config, field_path, delete_origin_value, replace_record, key_transformation, expected_record
+    input_record,
+    config,
+    field_path,
+    delete_origin_value,
+    replace_record,
+    key_transformation,
+    expected_record,
 ):
     flattener = DpathFlattenFields(
         field_path=field_path,