[Scalabrese] Added client_session_keep_alive = True to Snowflake

vanna-ai · Apr 13, 2024 · f4a6f42 · f4a6f42
1 parent 9a0adde
commit f4a6f42
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py
@@ -62,7 +62,6 @@
 import plotly.express as px
 import plotly.graph_objects as go
 import requests
-from snowflake.connector.connection import SnowflakeConnection
 
 from ..exceptions import DependencyError, ImproperlyConfigured, ValidationError
 from ..types import TrainingPlan, TrainingPlanItem
@@ -650,7 +649,7 @@ def connect_to_snowflake(
             else:
                 raise ImproperlyConfigured("Please set your Snowflake database.")
 
-        conn:SnowflakeConnection = snowflake.connector.connect(
+        conn = snowflake.connector.connect(
             user=username,
             password=password,
             account=account,