fix(ingest/snowflake): make test connection logs less noisy (#10587)

datahub-project · May 30, 2024 · 012930a · 012930a
1 parent db965d6
commit 012930a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/snowflake/snowflake_v2.py b/metadata-ingestion/src/datahub/ingestion/source/snowflake/snowflake_v2.py
@@ -103,6 +103,7 @@
 from datahub.ingestion.source.state.stateful_ingestion_base import (
     StatefulIngestionSourceBase,
 )
+from datahub.ingestion.source_config.sql.snowflake import BaseSnowflakeConfig
 from datahub.ingestion.source_report.ingestion_stage import (
     LINEAGE_EXTRACTION,
     METADATA_EXTRACTION,
@@ -328,7 +329,7 @@ def test_connection(config_dict: dict) -> TestConnectionReport:
         test_report = TestConnectionReport()
 
         try:
-            connection_conf = SnowflakeV2Config.parse_obj_allow_extras(config_dict)
+            connection_conf = BaseSnowflakeConfig.parse_obj_allow_extras(config_dict)
 
             connection: SnowflakeConnection = connection_conf.get_connection()
             assert connection
@@ -354,7 +355,7 @@ def test_connection(config_dict: dict) -> TestConnectionReport:
 
     @staticmethod
     def check_capabilities(
-        conn: SnowflakeConnection, connection_conf: SnowflakeV2Config
+        conn: SnowflakeConnection, connection_conf: BaseSnowflakeConfig
     ) -> Dict[Union[SourceCapability, str], CapabilityReport]:
         # Currently only overall capabilities are reported.
         # Resource level variations in capabilities are not considered.