chore: additional Python syntax modernization with pyupgrade

Co-Authored-By: Aaron <AJ> Steers <[email protected]>

Author: devin-ai-integration[bot]

airbyte_cdk/config_observation.py

@@ -57,7 +57,7 @@ def __setitem__(self, item: Any, value: Any) -> None:
             for i, sub_value in enumerate(value):
                 if isinstance(sub_value, MutableMapping):
                     value[i] = ObservedDict(sub_value, self.observer)
-        super(ObservedDict, self).__setitem__(item, value)
+        super().__setitem__(item, value)
         if self.update_on_unchanged_value or value != previous_value:
             self.observer.update()
 

airbyte_cdk/sources/declarative/concurrent_declarative_source.py

@@ -142,9 +142,9 @@ def read(
         # the concurrent streams must be saved so that they can be removed from the catalog before starting
         # synchronous streams
         if len(concurrent_streams) > 0:
-            concurrent_stream_names = set(
-                [concurrent_stream.name for concurrent_stream in concurrent_streams]
-            )
+            concurrent_stream_names = {
+                concurrent_stream.name for concurrent_stream in concurrent_streams
+            }
 
             selected_concurrent_streams = self._select_streams(
                 streams=concurrent_streams, configured_catalog=catalog

airbyte_cdk/sources/declarative/decoders/composite_raw_decoder.py

@@ -120,8 +120,7 @@ def parse(self, data: BufferedIOBase) -> PARSER_OUTPUT_TYPE:
         """
         text_data = TextIOWrapper(data, encoding=self.encoding)  # type: ignore
         reader = csv.DictReader(text_data, delimiter=self._get_delimiter() or ",")
-        for row in reader:
-            yield row
+        yield from reader
 
 
 class CompositeRawDecoder(Decoder):

airbyte_cdk/sources/declarative/extractors/response_to_file_extractor.py

@@ -142,8 +142,7 @@ def _read_with_chunks(
                 )
                 for chunk in chunks:
                     chunk = chunk.replace({nan: None}).to_dict(orient="records")
-                    for row in chunk:
-                        yield row
+                    yield from chunk
         except pd.errors.EmptyDataError as e:
             self.logger.info(f"Empty data received. {e}")
             yield from []

airbyte_cdk/sources/file_based/exceptions.py

@@ -63,7 +63,7 @@ def collect(self, logged_error: AirbyteMessage) -> None:
 
 
 class BaseFileBasedSourceError(Exception):
-    def __init__(self, error: Union[FileBasedSourceError, str], **kwargs):  # type: ignore # noqa
+    def __init__(self, error: FileBasedSourceError | str, **kwargs):  # type: ignore # noqa
         if isinstance(error, FileBasedSourceError):
             error = FileBasedSourceError(error).value
         super().__init__(

airbyte_cdk/sql/shared/sql_processor.py

@@ -210,7 +210,7 @@ def get_sql_engine(self) -> Engine:
         return self.sql_config.get_sql_engine()
 
     @contextmanager
-    def get_sql_connection(self) -> Generator[sqlalchemy.engine.Connection, None, None]:
+    def get_sql_connection(self) -> Generator[sqlalchemy.engine.Connection]:
         """A context manager which returns a new SQL connection for running queries.
 
         If the connection needs to close, it will be closed automatically.

airbyte_cdk/utils/schema_inferrer.py

@@ -32,7 +32,7 @@ class NoRequiredObj(Object):
     """
 
     def to_schema(self) -> Mapping[str, Any]:
-        schema: dict[str, Any] = super(NoRequiredObj, self).to_schema()
+        schema: dict[str, Any] = super().to_schema()
         schema.pop("required", None)
         return schema
 

unit_tests/sources/declarative/interpolation/test_filters.py

@@ -42,7 +42,7 @@ def test_hash_md5_with_salt() -> None:
     input_string = "test_input_string"
     input_salt = "test_input_salt"
 
-    s = "{{ '%s' | hash('md5', '%s' ) }}" % (input_string, input_salt)
+    s = "{{{{ '{}' | hash('md5', '{}' ) }}}}".format(input_string, input_salt)
     filter_hash = interpolation.eval(s, config={})
 
     # compute expected value calling hashlib directly
@@ -112,7 +112,7 @@ def test_hmac_sha256_default() -> None:
     message = "test_message"
     secret_key = "test_secret_key"
 
-    s = "{{ '%s' | hmac('%s') }}" % (message, secret_key)
+    s = "{{{{ '{}' | hmac('{}') }}}}".format(message, secret_key)
     filter_hmac = interpolation.eval(s, config={})
 
     # compute expected hmac using the hmac library directly
@@ -128,7 +128,7 @@ def test_hmac_sha256_explicit() -> None:
     message = "test_message"
     secret_key = "test_secret_key"
 
-    s = "{{ '%s' | hmac('%s', 'sha256') }}" % (message, secret_key)
+    s = "{{{{ '{}' | hmac('{}', 'sha256') }}}}".format(message, secret_key)
     filter_hmac = interpolation.eval(s, config={})
 
     # compute expected hmac using the hmac library directly
@@ -160,7 +160,7 @@ def test_hmac_with_invalid_hash_type() -> None:
     message = "test_message"
     secret_key = "test_secret_key"
 
-    s = "{{ '%s' | hmac('%s', 'md5') }}" % (message, secret_key)
+    s = "{{{{ '{}' | hmac('{}', 'md5') }}}}".format(message, secret_key)
 
     with pytest.raises(ValueError):
         interpolation.eval(s, config={})

unit_tests/sources/declarative/requesters/error_handlers/backoff_strategies/test_wait_time_from_header.py

@@ -39,7 +39,7 @@
         ),
         ("test_wait_time_from_header_not_a_number", "wait_time", "61,60", None, None),
         ("test_wait_time_from_header_with_regex", "wait_time", "61,60", r"([-+]?\d+)", 61),  # noqa
-        ("test_wait_time_fœrom_header_with_regex_no_match", "wait_time", "...", "[-+]?\d+", None),  # noqa
+        ("test_wait_time_fœrom_header_with_regex_no_match", "wait_time", "...", r"[-+]?\d+", None),  # noqa
         ("test_wait_time_from_header", "absent_header", None, None, None),
     ],
 )

unit_tests/sources/declarative/requesters/error_handlers/backoff_strategies/test_wait_until_time_from_header.py

@@ -53,7 +53,7 @@
             "wait_until",
             "1600000060,60",
             None,
-            "[-+]?\d+",
+            r"[-+]?\d+",
             60,
         ),  # noqa
         (
@@ -78,7 +78,7 @@
             "wait_time",
             "...",
             None,
-            "[-+]?\d+",
+            r"[-+]?\d+",
             None,
         ),  # noqa
         (