import protocol models from airbyte-protocol library

airbytehq · Sep 23, 2024 · c36751d · c36751d
1 parent d1429b3
commit c36751d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/integration_tests/destinations/test_source_to_destination.py b/tests/integration_tests/destinations/test_source_to_destination.py
@@ -16,7 +16,7 @@
 from airbyte.shared.catalog_providers import CatalogProvider
 from airbyte.sources.base import Source
 from airbyte.strategies import WriteStrategy
-from airbyte_cdk import AirbyteMessage, AirbyteRecordMessage, Type
+from airbyte_protocol.models import AirbyteMessage, AirbyteRecordMessage, Type
 
 
 @pytest.fixture

diff --git a/tests/unit_tests/test_progress.py b/tests/unit_tests/test_progress.py
@@ -12,7 +12,7 @@
  _get_elapsed_time_str,
  _to_time_str,
 )
-from airbyte_cdk import AirbyteMessage, AirbyteRecordMessage, Type
+from airbyte_protocol.models import AirbyteMessage, AirbyteRecordMessage, Type
 from dateutil.tz import tzlocal
 from freezegun import freeze_time
 from rich.errors import LiveError