format

Author: maxime.c

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

@@ -1748,7 +1748,11 @@ def create_custom_component(self, model: Any, config: Config, **kwargs: Any) ->
 
             if self._is_component(model_value):
                 model_args[model_field] = self._create_nested_component(
-                    model, model_field, model_value, config, **kwargs,
+                    model,
+                    model_field,
+                    model_value,
+                    config,
+                    **kwargs,
                 )
             elif isinstance(model_value, list):
                 vals = []
@@ -1760,7 +1764,15 @@ def create_custom_component(self, model: Any, config: Config, **kwargs: Any) ->
                         if derived_type:
                             v["type"] = derived_type
                     if self._is_component(v):
-                        vals.append(self._create_nested_component(model, model_field, v, config, **kwargs,))
+                        vals.append(
+                            self._create_nested_component(
+                                model,
+                                model_field,
+                                v,
+                                config,
+                                **kwargs,
+                            )
+                        )
                     else:
                         vals.append(v)
                 model_args[model_field] = vals
@@ -2525,7 +2537,9 @@ def create_dynamic_schema_loader(
             config=config,
             name=name,
             primary_key=None,
-            partition_router=self._build_stream_slicer_from_partition_router(model.retriever, config),
+            partition_router=self._build_stream_slicer_from_partition_router(
+                model.retriever, config
+            ),
             transformations=[],
             use_cache=True,
             log_formatter=(
@@ -3238,7 +3252,9 @@ def _get_url(req: Requester) -> str:
 
         if not request_options_provider:
             request_options_provider = DefaultRequestOptionsProvider(parameters={})
-        if isinstance(request_options_provider, DefaultRequestOptionsProvider) and isinstance(partition_router, PartitionRouter):
+        if isinstance(request_options_provider, DefaultRequestOptionsProvider) and isinstance(
+            partition_router, PartitionRouter
+        ):
             request_options_provider = partition_router
 
         paginator = (
@@ -3705,9 +3721,7 @@ def create_parent_stream_config_with_substream_wrapper(
         self, model: ParentStreamConfigModel, config: Config, *, stream_name: str, **kwargs: Any
     ) -> Any:
         # getting the parent state
-        child_state = self._connector_state_manager.get_stream_state(
-            stream_name, None
-        )
+        child_state = self._connector_state_manager.get_stream_state(stream_name, None)
 
         # This flag will be used exclusively for StateDelegatingStream when a parent stream is created
         has_parent_state = bool(

unit_tests/sources/declarative/incremental/test_concurrent_perpartitioncursor.py

@@ -2133,9 +2133,7 @@ def test_incremental_parent_state_migration(
                         "states": [
                             {
                                 "partition": {"id": 1, "parent_slice": {}},
-                                "cursor": {
-                                    "updated_at": START_DATE
-                                },
+                                "cursor": {"updated_at": START_DATE},
                             }
                         ],
                         "lookback_window": 0,

unit_tests/sources/declarative/parsers/test_model_to_component_factory.py

@@ -2106,12 +2106,20 @@ def test_custom_components_do_not_contain_extra_fields():
     }
 
     custom_substream_partition_router = factory.create_component(
-        CustomPartitionRouterModel, custom_substream_partition_router_manifest, input_config, stream_name="child_stream_name",
+        CustomPartitionRouterModel,
+        custom_substream_partition_router_manifest,
+        input_config,
+        stream_name="child_stream_name",
     )
     assert isinstance(custom_substream_partition_router, TestingCustomSubstreamPartitionRouter)
 
     assert len(custom_substream_partition_router.parent_stream_configs) == 1
-    assert isinstance(custom_substream_partition_router.parent_stream_configs[0].stream.cursor._message_repository, StateFilteringMessageRepository)
+    assert isinstance(
+        custom_substream_partition_router.parent_stream_configs[
+            0
+        ].stream.cursor._message_repository,
+        StateFilteringMessageRepository,
+    )
     assert custom_substream_partition_router.parent_stream_configs[0].parent_key.eval({}) == "id"
     assert (
         custom_substream_partition_router.parent_stream_configs[0].partition_field.eval({})
@@ -2174,7 +2182,10 @@ def test_parse_custom_component_fields_if_subcomponent():
     }
 
     custom_substream_partition_router = factory.create_component(
-        CustomPartitionRouterModel, custom_substream_partition_router_manifest, input_config, stream_name="child_stream_name"
+        CustomPartitionRouterModel,
+        custom_substream_partition_router_manifest,
+        input_config,
+        stream_name="child_stream_name",
     )
     assert isinstance(custom_substream_partition_router, TestingCustomSubstreamPartitionRouter)
     assert custom_substream_partition_router.custom_field == "here"