From 01d3fc32aca1d13348b9f05e164b2d9e7f237a78 Mon Sep 17 00:00:00 2001
From: Miles Adkins <miles.adkins@snowflake.com>
Date: Sun, 14 Aug 2022 21:51:24 -0500
Subject: [PATCH] fix: Force Snowflake Session to Timezone UTC

Signed-off-by: Miles Adkins <miles.adkins@snowflake.com>
---
 sdk/python/feast/infra/utils/snowflake_utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/sdk/python/feast/infra/utils/snowflake_utils.py b/sdk/python/feast/infra/utils/snowflake_utils.py
index 6ba9e54489..6b3500b605 100644
--- 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)