Auto-fix lint and format issues

Author: octavia-squidington-iii

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

@@ -2125,7 +2125,9 @@ def _migrate_state(self, config, model):
         else:
             state_transformations = []
         stream_state = self.apply_stream_state_migrations(state_transformations, stream_state)
-        self._connector_state_manager.update_state_for_stream(stream_name=model.name, namespace=None, value=stream_state)
+        self._connector_state_manager.update_state_for_stream(
+            stream_name=model.name, namespace=None, value=stream_state
+        )
 
     def _is_stop_condition_on_cursor(self, model: DeclarativeStreamModel) -> bool:
         return bool(

unit_tests/sources/declarative/parsers/test_model_to_component_factory.py

@@ -3542,9 +3542,13 @@ def test_create_default_stream_with_datetime_cursor_then_runs_state_migrations()
             )
         ]
     )
-    factory = ModelToComponentFactory(emit_connector_builder_messages=True, connector_state_manager=connector_state_manager)
+    factory = ModelToComponentFactory(
+        emit_connector_builder_messages=True, connector_state_manager=connector_state_manager
+    )
     stream = factory.create_component(
-        model_type=DeclarativeStreamModel, component_definition=YamlDeclarativeSource._parse(content), config=input_config
+        model_type=DeclarativeStreamModel,
+        component_definition=YamlDeclarativeSource._parse(content),
+        config=input_config,
     )
     assert stream.cursor.state["states"] == [
         {"cursor": {"updated_at": stream_state["updated_at"]}, "partition": {"type": "type_1"}},
@@ -3627,9 +3631,7 @@ def test_create_concurrent_cursor_from_perpartition_cursor_runs_state_migrations
 
     stream_state = {
         "state": {"updated_at": "2025-01-01T00:00:00.000000+00:00"},
-        "parent_state": {
-            "parent_stream": {"updated_at": "2025-01-01T00:00:00.000000+00:00"}
-        }
+        "parent_state": {"parent_stream": {"updated_at": "2025-01-01T00:00:00.000000+00:00"}},
     }
     connector_state_manager = ConnectorStateManager(
         state=[
@@ -3642,14 +3644,19 @@ def test_create_concurrent_cursor_from_perpartition_cursor_runs_state_migrations
             )
         ]
     )
-    factory = ModelToComponentFactory(emit_connector_builder_messages=True,
-                                      connector_state_manager=connector_state_manager)
+    factory = ModelToComponentFactory(
+        emit_connector_builder_messages=True, connector_state_manager=connector_state_manager
+    )
     stream = factory.create_component(
-        model_type=DeclarativeStreamModel, component_definition=YamlDeclarativeSource._parse(content),
-        config=input_config
+        model_type=DeclarativeStreamModel,
+        component_definition=YamlDeclarativeSource._parse(content),
+        config=input_config,
     )
     assert stream.cursor.state["lookback_window"] == 20
-    assert stream.cursor._partition_router.parent_stream_configs[0].stream.cursor.state["updated_at"] == "2024-02-01T00:00:00.000000+0000"
+    assert (
+        stream.cursor._partition_router.parent_stream_configs[0].stream.cursor.state["updated_at"]
+        == "2024-02-01T00:00:00.000000+0000"
+    )
 
 
 def test_create_concurrent_cursor_uses_min_max_datetime_format_if_defined():

unit_tests/sources/declarative/parsers/testing_components.py

@@ -3,7 +3,7 @@
 #
 
 from dataclasses import dataclass, field
-from typing import ClassVar, List, Optional, Any, Mapping
+from typing import Any, ClassVar, List, Mapping, Optional
 
 from airbyte_cdk.sources.declarative.extractors import DpathExtractor
 from airbyte_cdk.sources.declarative.migrations.state_migration import StateMigration
@@ -78,5 +78,7 @@ def should_migrate(self, stream_state: Mapping[str, Any]) -> bool:
 
     def migrate(self, stream_state: Mapping[str, Any]) -> Mapping[str, Any]:
         stream_state["lookback_window"] = 20
-        stream_state["parent_state"]["parent_stream"] = {"updated_at": "2024-02-01T00:00:00.000000+00:00"}
+        stream_state["parent_state"]["parent_stream"] = {
+            "updated_at": "2024-02-01T00:00:00.000000+00:00"
+        }
         return stream_state