Fix

Signed-off-by: Kevin Zhang <[email protected]>
feast-dev · Apr 4, 2022 · 9ff0ac1 · 9ff0ac1
1 parent b96a3ab
commit 9ff0ac1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/sdk/python/feast/infra/offline_stores/snowflake_source.py b/sdk/python/feast/infra/offline_stores/snowflake_source.py
@@ -57,7 +57,11 @@ def __init__(
         _schema = "PUBLIC" if (database and table and not schema) else schema
 
         self.snowflake_options = SnowflakeOptions(
-            database=database, schema=_schema, table=table, query=query, warehouse=warehouse
+            database=database,
+            schema=_schema,
+            table=table,
+            query=query,
+            warehouse=warehouse,
         )
 
         # If no name, use the table as the default name
@@ -301,7 +305,7 @@ def warehouse(self):
         """Returns the warehouse name of this snowflake table."""
         return self._warehouse
 
-    @table.setter
+    @warehouse.setter
     def warehouse(self, warehouse):
         """Sets the warehouse name of this snowflake table."""
         self._warehouse = warehouse

diff --git a/sdk/python/feast/templates/snowflake/bootstrap.py b/sdk/python/feast/templates/snowflake/bootstrap.py
@@ -81,8 +81,6 @@ def bootstrap():
     replace_str_in_file(driver_file, "SNOWFLAKE_WAREHOUSE", snowflake_warehouse)
 
 
-
-
 def replace_str_in_file(file_path, match_str, sub_str):
     with open(file_path, "r") as f:
         contents = f.read()