Skip to content

Commit

Permalink
SAASMLOPS-1058-2 Support s3gov schema by snowflake offline store duri…
Browse files Browse the repository at this point in the history
…ng materialization

Signed-off-by: Alex Vinnik <[email protected]>
  • Loading branch information
alex-vinnik-sp committed Jan 17, 2024
1 parent ac6529c commit 30312dc
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion sdk/python/feast/infra/offline_stores/snowflake.py
Original file line number Diff line number Diff line change
Expand Up @@ -588,8 +588,10 @@ def to_remote_storage(self) -> List[str]:
file_name_column_index = [
idx for idx, rm in enumerate(cursor.description) if rm.name == "FILE_NAME"
][0]
# s3gov schema is used by Snowflake in AWS govcloud regions
native_export_path = self.export_path.replace("s3gov://", "s3://")
return [
f"{self.export_path}/{row[file_name_column_index]}"
f"{native_export_path}/{row[file_name_column_index]}"
for row in cursor.fetchall()
]

Expand Down

0 comments on commit 30312dc

Please sign in to comment.