chore: add module and __all__ imports

airbytehq · Oct 6, 2024 · f5ef21e · f5ef21e
1 parent d8d23b0
commit f5ef21e
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 1 deletion.
diff --git a/airbyte/destinations/__init__.py b/airbyte/destinations/__init__.py
@@ -79,6 +79,7 @@
 from airbyte.destinations.base import Destination
 from airbyte.destinations.util import (
     get_destination,
+    get_noop_destination,
 )
 
 
@@ -87,6 +88,7 @@
     "util",
     # Methods
     "get_destination",
+    "get_noop_destination",
     # Classes
     "Destination",
 ]
diff --git a/airbyte/destinations/util.py b/airbyte/destinations/util.py
@@ -89,3 +89,9 @@ def get_noop_destination() -> Destination:
         },
         docker_image=True,
     )
+
+
+__all__ = [
+    "get_destination",
+    "get_noop_destination",
+]
diff --git a/airbyte/sources/__init__.py b/airbyte/sources/__init__.py
@@ -10,7 +10,10 @@
     get_available_connectors,
     get_connector_metadata,
 )
-from airbyte.sources.util import get_source
+from airbyte.sources.util import (
+    get_benchmark_source,
+    get_source,
+)
 
 
 __all__ = [
@@ -19,6 +22,7 @@
     "util",
     # Factories
     "get_source",
+    "get_benchmark_source",
     # Helper Functions
     "get_available_connectors",
     "get_connector_metadata",

diff --git a/airbyte/sources/util.py b/airbyte/sources/util.py
@@ -165,4 +165,5 @@ def get_benchmark_source(
 
 __all__ = [
     "get_source",
+    "get_benchmark_source",
 ]