Skip to content

Commit 6dd571b

Browse files
committed
Fix snowflake
1 parent 9b6a6a3 commit 6dd571b

2 files changed

Lines changed: 8 additions & 0 deletions

File tree

sqlmesh/core/engine_adapter/snowflake.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -451,6 +451,10 @@ def _get_data_objects(
451451
if object_names:
452452
query = query.where(exp.column("TABLE_NAME").isin(*object_names))
453453

454+
# exclude SNOWPARK_TEMP_TABLE tables that are managed by the Snowpark library and are an implementation
455+
# detail of dealing with DataFrame's
456+
query = query.where(exp.column("TABLE_NAME").like("SNOWPARK_TEMP_TABLE%").not_())
457+
454458
df = self.fetchdf(query, quote_identifiers=True)
455459
if df.empty:
456460
return []

tests/core/engine_adapter/integration/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -713,6 +713,10 @@ def cleanup(self, ctx: t.Optional[Context] = None):
713713
schema_name=schema_name, ignore_if_not_exists=True, cascade=True
714714
)
715715

716+
if snowpark := self.engine_adapter.snowpark:
717+
# ensure that the next test gets a fresh Snowpark session
718+
snowpark.close()
719+
716720
def upsert_sql_model(self, model_definition: str) -> t.Tuple[Context, SqlModel]:
717721
if not self._context:
718722
self._context = self.create_context()

0 commit comments

Comments
 (0)