fix: apply ruff 0.11.5 formatting to resolve CI format check discrepancy

devin-ai-integration[bot] · aaronsteers · devin-ai-integration[bot] · commit aaee7398648e · 2025-08-29T03:57:15.000Z
Co-Authored-By: AJ Steers &lt;aj@airbyte.io&gt;
diff --git a/airbyte_cdk/connector_builder/main.py b/airbyte_cdk/connector_builder/main.py
@@ -80,9 +80,9 @@ def handle_connector_builder_request(
     if command == "resolve_manifest":
         return resolve_manifest(source)
     elif command == "test_read":
-        assert (
-            catalog is not None
-        ), "`test_read` requires a valid `ConfiguredAirbyteCatalog`, got None."
+        assert catalog is not None, (
+            "`test_read` requires a valid `ConfiguredAirbyteCatalog`, got None."
+        )
         return read_stream(source, config, catalog, state, limits)
     elif command == "full_resolve_manifest":
         return full_resolve_manifest(source, limits)
diff --git a/airbyte_cdk/sources/concurrent_source/concurrent_source.py b/airbyte_cdk/sources/concurrent_source/concurrent_source.py
@@ -51,9 +51,9 @@ def create(
         too_many_generator = (
             not is_single_threaded and initial_number_of_partitions_to_generate >= num_workers
         )
-        assert (
-            not too_many_generator
-        ), "It is required to have more workers than threads generating partitions"
+        assert not too_many_generator, (
+            "It is required to have more workers than threads generating partitions"
+        )
         threadpool = ThreadPoolManager(
             concurrent.futures.ThreadPoolExecutor(
                 max_workers=num_workers, thread_name_prefix="workerpool"
diff --git a/airbyte_cdk/sources/file_based/file_based_source.py b/airbyte_cdk/sources/file_based/file_based_source.py
@@ -282,9 +282,9 @@ def streams(self, config: Mapping[str, Any]) -> List[Stream]:
                     and hasattr(self, "_concurrency_level")
                     and self._concurrency_level is not None
                 ):
-                    assert (
-                        state_manager is not None
-                    ), "No ConnectorStateManager was created, but it is required for incremental syncs. This is unexpected. Please contact Support."
+                    assert state_manager is not None, (
+                        "No ConnectorStateManager was created, but it is required for incremental syncs. This is unexpected. Please contact Support."
+                    )
 
                     cursor = self.cursor_cls(
                         stream_config,
diff --git a/airbyte_cdk/sources/file_based/stream/concurrent/adapters.py b/airbyte_cdk/sources/file_based/stream/concurrent/adapters.py
@@ -277,9 +277,9 @@ def read(self) -> Iterable[Record]:
     def to_slice(self) -> Optional[Mapping[str, Any]]:
         if self._slice is None:
             return None
-        assert (
-            len(self._slice["files"]) == 1
-        ), f"Expected 1 file per partition but got {len(self._slice['files'])} for stream {self.stream_name()}"
+        assert len(self._slice["files"]) == 1, (
+            f"Expected 1 file per partition but got {len(self._slice['files'])} for stream {self.stream_name()}"
+        )
         file = self._slice["files"][0]
         return {"files": [file]}
 
diff --git a/airbyte_cdk/sql/shared/sql_processor.py b/airbyte_cdk/sql/shared/sql_processor.py
@@ -326,9 +326,9 @@ def _ensure_schema_exists(
 
         if DEBUG_MODE:
             found_schemas = schemas_list
-            assert (
-                schema_name in found_schemas
-            ), f"Schema {schema_name} was not created. Found: {found_schemas}"
+            assert schema_name in found_schemas, (
+                f"Schema {schema_name} was not created. Found: {found_schemas}"
+            )
 
     def _quote_identifier(self, identifier: str) -> str:
         """Return the given identifier, quoted."""
diff --git a/unit_tests/connector_builder/test_connector_builder_handler.py b/unit_tests/connector_builder/test_connector_builder_handler.py
@@ -1267,9 +1267,9 @@ def test_handle_read_external_requests(deployment_mode, url_base, expected_error
             source, config, catalog, _A_PER_PARTITION_STATE, limits
         ).record.data
         if expected_error:
-            assert (
-                len(output_data["logs"]) > 0
-            ), "Expected at least one log message with the expected error"
+            assert len(output_data["logs"]) > 0, (
+                "Expected at least one log message with the expected error"
+            )
             error_message = output_data["logs"][0]
             assert error_message["level"] == "ERROR"
             assert expected_error in error_message["stacktrace"]
@@ -1363,9 +1363,9 @@ def test_handle_read_external_oauth_request(deployment_mode, token_url, expected
             source, config, catalog, _A_PER_PARTITION_STATE, limits
         ).record.data
         if expected_error:
-            assert (
-                len(output_data["logs"]) > 0
-            ), "Expected at least one log message with the expected error"
+            assert len(output_data["logs"]) > 0, (
+                "Expected at least one log message with the expected error"
+            )
             error_message = output_data["logs"][0]
             assert error_message["level"] == "ERROR"
             assert expected_error in error_message["stacktrace"]
diff --git a/unit_tests/destinations/test_destination.py b/unit_tests/destinations/test_destination.py
@@ -58,9 +58,9 @@ def test_successful_parse(
         self, arg_list: List[str], expected_output: Mapping[str, Any], destination: Destination
     ):
         parsed_args = vars(destination.parse_args(arg_list))
-        assert (
-            parsed_args == expected_output
-        ), f"Expected parsing {arg_list} to return parsed args {expected_output} but instead found {parsed_args}"
+        assert parsed_args == expected_output, (
+            f"Expected parsing {arg_list} to return parsed args {expected_output} but instead found {parsed_args}"
+        )
 
     @pytest.mark.parametrize(
         ("arg_list"),
diff --git a/unit_tests/legacy/sources/declarative/partition_routers/test_parent_state_stream.py b/unit_tests/legacy/sources/declarative/partition_routers/test_parent_state_stream.py
@@ -340,10 +340,11 @@ def run_incremental_parent_state_test(
             expected_records_set = list(
                 {orjson.dumps(record): record for record in expected_records}.values()
             )
-            assert (
-                sorted(cumulative_records_state_deduped, key=lambda x: orjson.dumps(x))
-                == sorted(expected_records_set, key=lambda x: orjson.dumps(x))
-            ), f"Records mismatch with intermediate state {state}. Expected {expected_records}, got {cumulative_records_state_deduped}"
+            assert sorted(
+                cumulative_records_state_deduped, key=lambda x: orjson.dumps(x)
+            ) == sorted(expected_records_set, key=lambda x: orjson.dumps(x)), (
+                f"Records mismatch with intermediate state {state}. Expected {expected_records}, got {cumulative_records_state_deduped}"
+            )
 
             # Store the final state after each intermediate read
             final_state_intermediate = [
@@ -355,9 +356,9 @@ def run_incremental_parent_state_test(
 
         # Assert that the final state matches the expected state for all runs
         for i, final_state in enumerate(final_states):
-            assert (
-                final_state in expected_states
-            ), f"Final state mismatch at run {i + 1}. Expected {expected_states}, got {final_state}"
+            assert final_state in expected_states, (
+                f"Final state mismatch at run {i + 1}. Expected {expected_states}, got {final_state}"
+            )
 
 
 @pytest.mark.parametrize(
diff --git a/unit_tests/source_declarative_manifest/test_source_declarative_w_custom_components.py b/unit_tests/source_declarative_manifest/test_source_declarative_w_custom_components.py
@@ -97,9 +97,9 @@ def get_py_components_config_dict(
 
     manifest_dict = yaml.safe_load(manifest_yaml_path.read_text())
     assert manifest_dict, "Failed to load the manifest file."
-    assert isinstance(
-        manifest_dict, Mapping
-    ), f"Manifest file is type {type(manifest_dict).__name__}, not a mapping: {manifest_dict}"
+    assert isinstance(manifest_dict, Mapping), (
+        f"Manifest file is type {type(manifest_dict).__name__}, not a mapping: {manifest_dict}"
+    )
 
     custom_py_code = custom_py_code_path.read_text()
     combined_config_dict = {
diff --git a/unit_tests/sources/declarative/incremental/test_concurrent_perpartitioncursor.py b/unit_tests/sources/declarative/incremental/test_concurrent_perpartitioncursor.py
@@ -389,9 +389,9 @@ def run_mocked_test(
             request_count = len(
                 [req for req in m.request_history if unquote(req.url) == unquote(url)]
             )
-            assert (
-                request_count == 1
-            ), f"URL {url} was called {request_count} times, expected exactly once."
+            assert request_count == 1, (
+                f"URL {url} was called {request_count} times, expected exactly once."
+            )
 
 
 def _run_read(
@@ -1137,10 +1137,11 @@ def run_incremental_parent_state_test(
             expected_records_set = list(
                 {orjson.dumps(record): record for record in expected_records}.values()
             )
-            assert (
-                sorted(cumulative_records_state_deduped, key=lambda x: x["id"])
-                == sorted(expected_records_set, key=lambda x: x["id"])
-            ), f"Records mismatch with intermediate state {state}. Expected {expected_records}, got {cumulative_records_state_deduped}"
+            assert sorted(cumulative_records_state_deduped, key=lambda x: x["id"]) == sorted(
+                expected_records_set, key=lambda x: x["id"]
+            ), (
+                f"Records mismatch with intermediate state {state}. Expected {expected_records}, got {cumulative_records_state_deduped}"
+            )
 
             # Store the final state after each intermediate read
             final_state_intermediate = [
@@ -1151,9 +1152,9 @@ def run_incremental_parent_state_test(
 
         # Assert that the final state matches the expected state for all runs
         for i, final_state in enumerate(final_states):
-            assert (
-                final_state in expected_states
-            ), f"Final state mismatch at run {i + 1}. Expected {expected_states}, got {final_state}"
+            assert final_state in expected_states, (
+                f"Final state mismatch at run {i + 1}. Expected {expected_states}, got {final_state}"
+            )
 
 
 @pytest.mark.parametrize(
diff --git a/unit_tests/sources/declarative/interpolation/test_interpolated_string.py b/unit_tests/sources/declarative/interpolation/test_interpolated_string.py
@@ -62,6 +62,6 @@ def test_parsing_record_data(test_name, input_string, record_value, expected_val
     if expected_value is None:
         assert val is None, f"Expected None for value {record_value} in test {test_name}"
     else:
-        assert float == type(
-            val
-        ), f"Expected float, got {type(val)} for value {val} in test {test_name}"
+        assert float == type(val), (
+            f"Expected float, got {type(val)} for value {val} in test {test_name}"
+        )
diff --git a/unit_tests/sources/declarative/parsers/test_model_to_component_factory.py b/unit_tests/sources/declarative/parsers/test_model_to_component_factory.py
@@ -3437,9 +3437,9 @@ def migrate(self, stream_state: Mapping[str, Any]) -> Mapping[str, Any]:
         stream_state_migrations=[DummyStateMigration()],
     )
     assert cursor.state["lookback_window"] != 10, "State migration wasn't called"
-    assert (
-        cursor.state["lookback_window"] == 20
-    ), "State migration was called, but actual state don't match expected"
+    assert cursor.state["lookback_window"] == 20, (
+        "State migration was called, but actual state don't match expected"
+    )
 
 
 def test_create_concurrent_cursor_uses_min_max_datetime_format_if_defined():
diff --git a/unit_tests/sources/declarative/partition_routers/test_grouping_partition_router.py b/unit_tests/sources/declarative/partition_routers/test_grouping_partition_router.py
@@ -269,9 +269,9 @@ def __next__(self):
         cursor_slice={},
         extra_fields={"name": ["Board 0", "Board 1"], "owner": ["User0", "User1"]},
     )
-    assert (
-        controlled_iter.yield_count == 2
-    ), "Only 2 slices should be yielded to form the first group"
+    assert controlled_iter.yield_count == 2, (
+        "Only 2 slices should be yielded to form the first group"
+    )
 
     # Get the second slice
     second_slice = next(slices_iter)
@@ -280,9 +280,9 @@ def __next__(self):
         cursor_slice={},
         extra_fields={"name": ["Board 2", "Board 3"], "owner": ["User2", "User3"]},
     )
-    assert (
-        controlled_iter.yield_count == 4
-    ), "Only 4 slices should be yielded up to the second group"
+    assert controlled_iter.yield_count == 4, (
+        "Only 4 slices should be yielded up to the second group"
+    )
 
     # Exhaust the iterator
     remaining_slices = list(slices_iter)
@@ -293,9 +293,9 @@ def __next__(self):
             extra_fields={"name": ["Board 4"], "owner": ["User4"]},
         )
     ]
-    assert (
-        controlled_iter.yield_count == 5
-    ), "All 5 slices should be yielded after exhausting the iterator"
+    assert controlled_iter.yield_count == 5, (
+        "All 5 slices should be yielded after exhausting the iterator"
+    )
 
 
 def test_set_initial_state_delegation(mock_config, mock_underlying_router):
diff --git a/unit_tests/sources/declarative/partition_routers/test_substream_partition_router.py b/unit_tests/sources/declarative/partition_routers/test_substream_partition_router.py
@@ -1118,13 +1118,13 @@ def test_substream_using_resumable_full_refresh_parent_stream_slices(use_increme
     # validate final state for closed substream slices
     final_state = substream_cursor_slicer.get_stream_state()
     if not use_incremental_dependency:
-        assert (
-            final_state["states"] == expected_substream_state["states"]
-        ), "State for substreams is not valid!"
+        assert final_state["states"] == expected_substream_state["states"], (
+            "State for substreams is not valid!"
+        )
     else:
-        assert (
-            final_state == expected_substream_state
-        ), "State for substreams with incremental dependency is not valid!"
+        assert final_state == expected_substream_state, (
+            "State for substreams with incremental dependency is not valid!"
+        )
 
 
 @pytest.mark.parametrize(
diff --git a/unit_tests/sources/declarative/test_concurrent_declarative_source.py b/unit_tests/sources/declarative/test_concurrent_declarative_source.py
@@ -1471,9 +1471,9 @@ def test_concurrent_declarative_source_runs_state_migrations_provided_in_manifes
         source_config=manifest, config=_CONFIG, catalog=_CATALOG, state=state
     )
     concurrent_streams, synchronous_streams = source._group_streams(_CONFIG)
-    assert (
-        concurrent_streams[0].cursor.state.get("state") != state_blob.__dict__
-    ), "State was not migrated."
+    assert concurrent_streams[0].cursor.state.get("state") != state_blob.__dict__, (
+        "State was not migrated."
+    )
     assert concurrent_streams[0].cursor.state.get("states") == [
         {"cursor": {"updated_at": "2024-08-21"}, "partition": {"type": "type_1"}},
         {"cursor": {"updated_at": "2024-08-21"}, "partition": {"type": "type_2"}},
diff --git a/unit_tests/sources/file_based/test_scenarios.py b/unit_tests/sources/file_based/test_scenarios.py
@@ -122,9 +122,9 @@ def _verify_read_output(output: EntrypointOutput, scenario: TestScenario[Abstrac
 
     expected_states = list(filter(lambda e: "data" not in e, expected_records))
     states = list(filter(lambda r: r.state, records_and_state_messages))
-    assert (
-        len(states) > 0
-    ), "No state messages emitted. Successful syncs should emit at least one stream state."
+    assert len(states) > 0, (
+        "No state messages emitted. Successful syncs should emit at least one stream state."
+    )
     _verify_state_record_counts(sorted_records, states)
 
     if hasattr(scenario.source, "cursor_cls") and issubclass(
@@ -182,9 +182,9 @@ def _verify_analytics(
     expected_analytics: Optional[List[AirbyteAnalyticsTraceMessage]],
 ) -> None:
     if expected_analytics:
-        assert (
-            len(analytics) == len(expected_analytics)
-        ), f"Number of actual analytics messages ({len(analytics)}) did not match expected ({len(expected_analytics)})"
+        assert len(analytics) == len(expected_analytics), (
+            f"Number of actual analytics messages ({len(analytics)}) did not match expected ({len(expected_analytics)})"
+        )
         for actual, expected in zip(analytics, expected_analytics):
             actual_type, actual_value = actual.trace.analytics.type, actual.trace.analytics.value
             expected_type = expected.type
diff --git a/unit_tests/sources/streams/test_call_rate.py b/unit_tests/sources/streams/test_call_rate.py
@@ -268,9 +268,9 @@ def test_limit_rate(self):
 
         with pytest.raises(CallRateLimitHit) as excinfo2:
             policy.try_acquire("call", weight=1), "call over limit"
-        assert (
-            excinfo2.value.time_to_wait < excinfo1.value.time_to_wait
-        ), "time to wait must decrease over time"
+        assert excinfo2.value.time_to_wait < excinfo1.value.time_to_wait, (
+            "time to wait must decrease over time"
+        )
 
     def test_limit_rate_support_custom_weight(self):
         """try_acquire must take into account provided weight and throw CallRateLimitHit when hit the limit."""
@@ -279,9 +279,9 @@ def test_limit_rate_support_custom_weight(self):
         policy.try_acquire("call", weight=2), "1st call with weight of 2"
         with pytest.raises(CallRateLimitHit) as excinfo:
             policy.try_acquire("call", weight=9), "2nd call, over limit since 2 + 9 = 11 > 10"
-        assert excinfo.value.time_to_wait.total_seconds() == pytest.approx(
-            60, 0.1
-        ), "should wait 1 minute before next call"
+        assert excinfo.value.time_to_wait.total_seconds() == pytest.approx(60, 0.1), (
+            "should wait 1 minute before next call"
+        )
 
     def test_multiple_limit_rates(self):
         """try_acquire must take into all call rates and apply stricter."""
diff --git a/unit_tests/sources/streams/test_stream_read.py b/unit_tests/sources/streams/test_stream_read.py
@@ -758,9 +758,9 @@ def test_configured_json_schema_with_invalid_properties():
     assert old_user_insights not in configured_json_schema_properties
     assert old_feature_info not in configured_json_schema_properties
     for stream_schema_property in stream_schema["properties"]:
-        assert (
-            stream_schema_property in configured_json_schema_properties
-        ), f"Stream schema property: {stream_schema_property} missing in configured schema"
+        assert stream_schema_property in configured_json_schema_properties, (
+            f"Stream schema property: {stream_schema_property} missing in configured schema"
+        )
         assert (
             stream_schema["properties"][stream_schema_property]
             == configured_json_schema_properties[stream_schema_property]
diff --git a/unit_tests/test_entrypoint.py b/unit_tests/test_entrypoint.py
@@ -830,9 +830,9 @@ def test_handle_record_counts(
         assert message_count == expected_records_by_stream[stream_descriptor]
 
     if actual_message.type == Type.STATE:
-        assert isinstance(
-            actual_message.state.sourceStats.recordCount, float
-        ), "recordCount value should be expressed as a float"
+        assert isinstance(actual_message.state.sourceStats.recordCount, float), (
+            "recordCount value should be expressed as a float"
+        )
 
 
 def test_given_serialization_error_using_orjson_then_fallback_on_json(
diff --git a/unit_tests/test_exception_handler.py b/unit_tests/test_exception_handler.py
diff --git a/unit_tests/test_secure_logger.py b/unit_tests/test_secure_logger.py
diff --git a/unit_tests/utils/test_secret_utils.py b/unit_tests/utils/test_secret_utils.py
