format/lint

Author: maxime.c

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

@@ -122,6 +122,9 @@
     DEPRECATION_LOGS_TAG,
     BaseModelWithDeprecations,
 )
+from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
+    Action1 as PaginationResetActionModel,
+)
 from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
     AddedFieldDefinition as AddedFieldDefinitionModel,
 )
@@ -362,9 +365,6 @@
 from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
     PaginationReset as PaginationResetModel,
 )
-from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
-    Action1 as PaginationResetActionModel,
-)
 from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
     ParametrizedComponentsResolver as ParametrizedComponentsResolverModel,
 )

airbyte_cdk/sources/streams/concurrent/cursor.py

@@ -555,7 +555,9 @@ def reduce_slice_range(self, stream_slice: StreamSlice) -> StreamSlice:
                     ]: self._connector_state_converter.output_format(
                         self._most_recent_cursor_value_per_partition[stream_slice]
                     ),
-                    self._slice_boundary_fields_wrapper[self._END_BOUNDARY]: stream_slice.cursor_slice[
+                    self._slice_boundary_fields_wrapper[
+                        self._END_BOUNDARY
+                    ]: stream_slice.cursor_slice[
                         self._slice_boundary_fields_wrapper[self._END_BOUNDARY]
                     ],
                 },

unit_tests/sources/declarative/test_concurrent_declarative_source.py

@@ -4947,8 +4947,11 @@ def test_given_pagination_reset_action_is_reset_even_though_stream_is_incrementa
             HttpResponse(
                 json.dumps(
                     {
-                        "results": [{"id": 1, "updated_at": "2022-02-01"}, {"id": 2, "updated_at": "2022-03-01"}],
-                        "next": "https://example.org/test?from=2022-01-01&cursor=toto"
+                        "results": [
+                            {"id": 1, "updated_at": "2022-02-01"},
+                            {"id": 2, "updated_at": "2022-03-01"},
+                        ],
+                        "next": "https://example.org/test?from=2022-01-01&cursor=toto",
                     }
                 ),
                 200,
@@ -4958,15 +4961,8 @@ def test_given_pagination_reset_action_is_reset_even_though_stream_is_incrementa
             HttpRequest("https://example.org/test?from=2022-01-01&cursor=toto"),
             [
                 HttpResponse(json.dumps({}), 400),
-                HttpResponse(
-                    json.dumps(
-                        {
-                            "results": [{"id": 3, "updated_at": "2022-04-01"}]
-                        }
-                    ),
-                    200
-                ),
-            ]
+                HttpResponse(json.dumps({"results": [{"id": 3, "updated_at": "2022-04-01"}]}), 200),
+            ],
         )
         messages = list(
             source.read(logger=source.logger, config=input_config, catalog=catalog, state=[])