Allow extra fields in Snowflake connections

Signed-off-by: michael neely <[email protected]>

Author: michaelneely

sdk/python/feast/infra/compute_engines/snowflake/snowflake_engine.py

@@ -87,7 +87,7 @@ class SnowflakeComputeEngineConfig(FeastConfigBaseModel):
 
     schema_: Optional[str] = Field("PUBLIC", alias="schema")
     """ Snowflake schema name """
-    model_config = ConfigDict(populate_by_name=True)
+    model_config = ConfigDict(populate_by_name=True, extra="allow")
 
 
 class SnowflakeComputeEngine(ComputeEngine):

sdk/python/feast/infra/offline_stores/snowflake.py

@@ -133,7 +133,7 @@ class SnowflakeOfflineStoreConfig(FeastConfigBaseModel):
 
     max_file_size: Optional[int] = None
     """ Upper size limit (in bytes) of each file that is offloaded. Default: 16777216"""
-    model_config = ConfigDict(populate_by_name=True)
+    model_config = ConfigDict(populate_by_name=True, extra="allow")
 
 
 class SnowflakeOfflineStore(OfflineStore):

sdk/python/feast/infra/online_stores/snowflake.py

@@ -67,7 +67,7 @@ class SnowflakeOnlineStoreConfig(FeastConfigBaseModel):
 
     schema_: Optional[str] = Field("PUBLIC", alias="schema")
     """ Snowflake schema name """
-    model_config = ConfigDict(populate_by_name=True)
+    model_config = ConfigDict(populate_by_name=True, extra="allow")
 
 
 class SnowflakeOnlineStore(OnlineStore):

sdk/python/feast/infra/registry/snowflake.py

@@ -117,7 +117,7 @@ class SnowflakeRegistryConfig(RegistryConfig):
 
     schema_: Optional[str] = Field("PUBLIC", alias="schema")
     """ Snowflake schema name """
-    model_config = ConfigDict(populate_by_name=True)
+    model_config = ConfigDict(populate_by_name=True, extra="allow")
 
 
 class SnowflakeRegistry(BaseRegistry):