diff --git a/snowflake_utils/models/table.py b/snowflake_utils/models/table.py index 3e96654..d5483b4 100644 --- a/snowflake_utils/models/table.py +++ b/snowflake_utils/models/table.py @@ -368,6 +368,7 @@ def merge( qualify: bool = False, files: list[str] | None = None, copy_grants: bool = True, + stage: str | None = None, ) -> None: def copy_callable(table: Table, sync_tags: bool) -> None: return table.copy_into( @@ -378,6 +379,7 @@ def copy_callable(table: Table, sync_tags: bool) -> None: sync_tags=sync_tags, files=files, copy_grants=copy_grants, + stage=stage, ) return self._merge(copy_callable, primary_keys, replication_keys, qualify)