Auto-fix lint and format issues

Author: octavia-squidington-iii

airbyte_cdk/sources/declarative/checks/check_stream.py

@@ -7,14 +7,16 @@
 from dataclasses import InitVar, dataclass
 from typing import Any, Dict, List, Mapping, Optional, Tuple, Union
 
-from airbyte_cdk.sources.streams.core import Stream
 from airbyte_cdk.sources.abstract_source import AbstractSource
 from airbyte_cdk.sources.declarative.checks.connection_checker import ConnectionChecker
 from airbyte_cdk.sources.streams.concurrent.abstract_stream import AbstractStream
+from airbyte_cdk.sources.streams.core import Stream
 from airbyte_cdk.sources.streams.http.availability_strategy import HttpAvailabilityStrategy
 
 
-def evaluate_availability(stream: Union[Stream, AbstractStream], logger: logging.Logger) -> Tuple[bool, Optional[str]]:
+def evaluate_availability(
+    stream: Union[Stream, AbstractStream], logger: logging.Logger
+) -> Tuple[bool, Optional[str]]:
     """
     As a transition period, we want to support both Stream and AbstractStream until we migrate everything to AbstractStream.
     """
@@ -97,7 +99,10 @@ def check_connection(
         return True, None
 
     def _check_stream_availability(
-        self, stream_name_to_stream: Dict[str, Union[Stream, AbstractStream]], stream_name: str, logger: logging.Logger
+        self,
+        stream_name_to_stream: Dict[str, Union[Stream, AbstractStream]],
+        stream_name: str,
+        logger: logging.Logger,
     ) -> Tuple[bool, Any]:
         """Checks if streams are available."""
         try:
@@ -112,7 +117,10 @@ def _check_stream_availability(
         return True, None
 
     def _check_dynamic_streams_availability(
-        self, source: AbstractSource, stream_name_to_stream: Dict[str, Union[Stream, AbstractStream]], logger: logging.Logger
+        self,
+        source: AbstractSource,
+        stream_name_to_stream: Dict[str, Union[Stream, AbstractStream]],
+        logger: logging.Logger,
     ) -> Tuple[bool, Any]:
         """Checks the availability of dynamic streams."""
         dynamic_streams = source.resolved_manifest.get("dynamic_streams", [])  # type: ignore[attr-defined] # The source's resolved_manifest manifest is checked before calling this method

airbyte_cdk/sources/streams/concurrent/availability_strategy.py

@@ -7,7 +7,6 @@
 
 
 class StreamAvailability:
-
     @classmethod
     def available(cls) -> "StreamAvailability":
         return cls(True)

airbyte_cdk/sources/streams/concurrent/default_stream.py

@@ -16,7 +16,6 @@
 
 
 class DefaultStream(AbstractStream):
-
     def __init__(
         self,
         partition_generator: PartitionGenerator,

unit_tests/sources/declarative/checks/test_check_stream.py

@@ -16,8 +16,8 @@
 from airbyte_cdk.sources.declarative.concurrent_declarative_source import (
     ConcurrentDeclarativeSource,
 )
-from airbyte_cdk.sources.streams.http import HttpStream
 from airbyte_cdk.sources.streams.core import Stream
+from airbyte_cdk.sources.streams.http import HttpStream
 from airbyte_cdk.test.mock_http import HttpMocker, HttpRequest, HttpResponse
 
 logger = logging.getLogger("test")

unit_tests/sources/streams/concurrent/test_default_stream.py

@@ -259,16 +259,22 @@ def test_given_no_partitions_when_get_availability_then_unavailable(self) -> Non
         assert availability.is_available == False
         assert "no stream slices were found" in availability.reason
 
-    def test_given_AirbyteTracedException_when_generating_partitions_when_get_availability_then_unavailable(self) -> None:
+    def test_given_AirbyteTracedException_when_generating_partitions_when_get_availability_then_unavailable(
+        self,
+    ) -> None:
         error_message = "error while generating partitions"
-        self._partition_generator.generate.side_effect = AirbyteTracedException(message=error_message)
+        self._partition_generator.generate.side_effect = AirbyteTracedException(
+            message=error_message
+        )
 
         availability = self._stream.check_availability()
 
         assert availability.is_available == False
         assert error_message in availability.reason
 
-    def test_given_unknown_error_when_generating_partitions_when_get_availability_then_raise(self) -> None:
+    def test_given_unknown_error_when_generating_partitions_when_get_availability_then_raise(
+        self,
+    ) -> None:
         """
         I'm not sure why we handle AirbyteTracedException but not other exceptions but this is to keep feature compatibility with HttpAvailabilityStrategy
         """
@@ -292,7 +298,9 @@ def test_given_records_when_get_availability_then_available(self) -> None:
 
         assert availability.is_available == True
 
-    def test_given_AirbyteTracedException_when_reading_records_when_get_availability_then_unavailable(self) -> None:
+    def test_given_AirbyteTracedException_when_reading_records_when_get_availability_then_unavailable(
+        self,
+    ) -> None:
         self._partition_generator.generate.return_value = [self._partition]
         error_message = "error while reading records"
         self._partition.read.side_effect = AirbyteTracedException(message=error_message)