diff --git a/airbyte/destinations/__init__.py b/airbyte/destinations/__init__.py index c35d1786..8a03ab75 100644 --- 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 index cd334337..b753e4d1 100644 --- 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 index 8af539e3..f29c6ea7 100644 --- 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 index 37977a29..4a9850bc 100644 --- a/airbyte/sources/util.py +++ b/airbyte/sources/util.py @@ -165,4 +165,5 @@ def get_benchmark_source( __all__ = [ "get_source", + "get_benchmark_source", ]