diff --git a/core/wren/src/wren/connector/datafusion.py b/core/wren/src/wren/connector/datafusion.py index da27d46f15..25c6afeafa 100644 --- a/core/wren/src/wren/connector/datafusion.py +++ b/core/wren/src/wren/connector/datafusion.py @@ -43,7 +43,7 @@ def dry_run(self, sql: str) -> None: # break the LIMIT subquery wrap (``SELECT 1;`` is a multi-statement batch # the planner rejects). Strip only the terminating run so ';' inside # string literals stays intact — same helper already used by ``query``. - self.ctx.dry_run(_strip_trailing_semicolon(sql)) + self.ctx.dry_run(strip_trailing_semicolon(sql)) def close(self) -> None: pass diff --git a/core/wren/src/wren/connector/snowflake.py b/core/wren/src/wren/connector/snowflake.py index 15d48bf0b1..37bbdfb48f 100644 --- a/core/wren/src/wren/connector/snowflake.py +++ b/core/wren/src/wren/connector/snowflake.py @@ -88,7 +88,7 @@ def query(self, sql: str, limit: int | None = None) -> pa.Table: def dry_run(self, sql: str) -> None: # ``describe`` still fails when the statement is terminated with ``;`` # (ProgrammingError: unexpected ';'). Strip only the trailing run. - cleaned = _strip_trailing_semicolon(sql) + cleaned = strip_trailing_semicolon(sql) try: with self.connection.cursor() as cursor: cursor.describe(cleaned)