fix import errors

airbytehq · Feb 17, 2024 · dbce864 · dbce864
1 parent 79eb7e1
commit dbce864
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/tests/integration_tests/test_snowflake_cache.py b/tests/integration_tests/test_snowflake_cache.py
@@ -9,13 +9,9 @@
 from collections.abc import Generator
 import os
 import sys
-import shutil
-from pathlib import Path
 
 import pytest
 
-from airbyte_cdk.models import ConfiguredAirbyteCatalog
-
 import airbyte as ab
 from airbyte import caches
 

diff --git a/tests/integration_tests/test_source_faker_integration.py b/tests/integration_tests/test_source_faker_integration.py
@@ -11,11 +11,11 @@
 import sys
 import shutil
 from pathlib import Path
+import typing
 
 import pytest
 import ulid
 
-from airbyte_cdk.models import ConfiguredAirbyteCatalog
 
 import airbyte as ab
 from airbyte import caches
@@ -130,7 +130,7 @@ def test_faker_pks(
 ) -> None:
     """Test that the append strategy works as expected."""
 
-    catalog: ConfiguredAirbyteCatalog = source_faker_seed_a.configured_catalog
+    catalog = source_faker_seed_a.configured_catalog
 
     assert catalog.streams[0].primary_key
     assert catalog.streams[1].primary_key