From dd6087452daa90d9dd4dd42ed2477d17345420da Mon Sep 17 00:00:00 2001 From: bindipankhudi Date: Fri, 10 May 2024 13:03:16 -0700 Subject: [PATCH] fixing unit tests --- .../unit_tests/snowflake_cortex_indexer_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/airbyte-integrations/connectors/destination-snowflake-cortex/unit_tests/snowflake_cortex_indexer_test.py b/airbyte-integrations/connectors/destination-snowflake-cortex/unit_tests/snowflake_cortex_indexer_test.py index 1993575dde1f..38cdac5e3453 100644 --- a/airbyte-integrations/connectors/destination-snowflake-cortex/unit_tests/snowflake_cortex_indexer_test.py +++ b/airbyte-integrations/connectors/destination-snowflake-cortex/unit_tests/snowflake_cortex_indexer_test.py @@ -2,7 +2,7 @@ # Copyright (c) 2023 Airbyte, Inc., all rights reserved. # -from typing import cast +from typing import cast, Optional from unittest.mock import ANY, MagicMock, Mock, call, patch from airbyte.strategies import WriteStrategy @@ -18,7 +18,7 @@ ) -def _create_snowflake_cortex_indexer(catalog:ConfiguredAirbyteCatalog | None = None ): +def _create_snowflake_cortex_indexer(catalog:Optional[ConfiguredAirbyteCatalog] ): config = SnowflakeCortexIndexingModel(account="account", username="username", password="password", database="database", warehouse="warehouse", role="role") with patch.object(SnowflakeCortexIndexer, '_init_db_connection', side_effect=None): indexer = SnowflakeCortexIndexer(config, 3, Mock(ConfiguredAirbyteCatalog) if catalog is None else catalog)