fix: Fixed bug in passing config file params to snowflake python connector (#2503)

sfc-gh-madkins · web-flow · commit 34f2b591bc8d · 2022-04-07T11:28:24.000-07:00
* fixed bug in passing config file params to snowflake python connector

Signed-off-by: Miles Adkins &lt;miles.adkins@snowflake.com&gt;

* removed .keys() bug

Signed-off-by: Miles Adkins &lt;miles.adkins@snowflake.com&gt;

* fixed schema assignment error

Signed-off-by: Miles Adkins &lt;miles.adkins@snowflake.com&gt;
diff --git a/sdk/python/feast/infra/offline_stores/snowflake.py b/sdk/python/feast/infra/offline_stores/snowflake.py
@@ -82,7 +82,7 @@ class SnowflakeOfflineStoreConfig(FeastConfigBaseModel):
     database: Optional[str] = None
     """ Snowflake database name """
 
-    schema_: Optional[str] = Field("PUBLIC", alias="schema")
+    schema_: Optional[str] = Field(None, alias="schema")
     """ Snowflake schema name """
 
     class Config:
diff --git a/sdk/python/feast/infra/utils/snowflake_utils.py b/sdk/python/feast/infra/utils/snowflake_utils.py
@@ -53,13 +53,19 @@ def get_snowflake_conn(config, autocommit=True) -> SnowflakeConnection:
     else:
         kwargs = {}
 
+    if "schema" in kwargs:
+        kwargs["schema_"] = kwargs.pop("schema")
+
     kwargs.update((k, v) for k, v in config_dict.items() if v is not None)
-    [
-        kwargs.update({k: '"' + v + '"'})
-        for k, v in kwargs.items()
-        if k in ["role", "warehouse", "database", "schema_"]
-    ]
-    kwargs["schema"] = kwargs.pop("schema_")
+
+    for k, v in kwargs.items():
+        if k in ["role", "warehouse", "database", "schema_"]:
+            kwargs[k] = f'"{v}"'
+
+    if "schema_" in kwargs:
+        kwargs["schema"] = kwargs.pop("schema_")
+    else:
+        kwargs["schema"] = '"PUBLIC"'
 
     try:
         conn = snowflake.connector.connect(
