fix: Force Snowflake Session to Timezone UTC

sfc-gh-madkins · sfc-gh-madkins · commit 01d3fc32aca1 · 2022-08-14T21:51:24.000-05:00
Signed-off-by: Miles Adkins &lt;miles.adkins@snowflake.com&gt;
diff --git a/sdk/python/feast/infra/utils/snowflake_utils.py b/sdk/python/feast/infra/utils/snowflake_utils.py
@@ -88,6 +88,8 @@ def get_snowflake_conn(config, autocommit=True) -> SnowflakeConnection:
             **kwargs,
         )
 
+        conn.cursor().execute("ALTER SESSION SET TIMEZONE = 'UTC'", _is_internal=True)
+
         return conn
     except KeyError as e:
         raise SnowflakeIncompleteConfig(e)

Original file line number	Diff line number	Diff line change
`@@ -88,6 +88,8 @@ def get_snowflake_conn(config, autocommit=True) -> SnowflakeConnection:`
`88`	`88`	`**kwargs,`
`89`	`89`	`)`
`90`	`90`
	`91`	`+ conn.cursor().execute("ALTER SESSION SET TIMEZONE = 'UTC'", _is_internal=True)`
	`92`	`+`
`91`	`93`	`return conn`
`92`	`94`	`except KeyError as e:`
`93`	`95`	`raise SnowflakeIncompleteConfig(e)`