diff --git a/sdk/python/feast/infra/offline_stores/snowflake_source.py b/sdk/python/feast/infra/offline_stores/snowflake_source.py index c15e5fe1a3..f094d2b329 100644 --- 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 index 6acbcab77d..194ba08c08 100644 --- 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()