From cecd9c3a46e4b66fb4ee66a009745f0dd22397c9 Mon Sep 17 00:00:00 2001 From: Chelsea Lin Date: Mon, 10 Nov 2025 18:26:53 +0000 Subject: [PATCH] feat: enable SQLGlot compiler as default --- bigframes/core/compile/__init__.py | 2 +- bigframes/session/direct_gbq_execution.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/bigframes/core/compile/__init__.py b/bigframes/core/compile/__init__.py index 68c36df288..f6b07e72b5 100644 --- a/bigframes/core/compile/__init__.py +++ b/bigframes/core/compile/__init__.py @@ -15,7 +15,7 @@ from bigframes.core.compile.api import test_only_ibis_inferred_schema from bigframes.core.compile.configs import CompileRequest, CompileResult -from bigframes.core.compile.ibis_compiler.ibis_compiler import compile_sql +from bigframes.core.compile.sqlglot.compiler import compile_sql __all__ = [ "test_only_ibis_inferred_schema", diff --git a/bigframes/session/direct_gbq_execution.py b/bigframes/session/direct_gbq_execution.py index 748c43e66c..9d8cc90fc1 100644 --- a/bigframes/session/direct_gbq_execution.py +++ b/bigframes/session/direct_gbq_execution.py @@ -20,7 +20,9 @@ import google.cloud.bigquery.table as bq_table from bigframes.core import compile, nodes -from bigframes.core.compile import sqlglot +from bigframes.core.compile.ibis_compiler.ibis_compiler import ( + compile_sql as compile_sql_ibis, +) import bigframes.core.events from bigframes.session import executor, semi_executor import bigframes.session._io.bigquery as bq_io @@ -40,7 +42,7 @@ def __init__( ): self.bqclient = bqclient self._compile_fn = ( - compile.compile_sql if compiler == "ibis" else sqlglot.compile_sql + compile_sql_ibis if compiler == "ibis" else compile.compile_sql ) self._publisher = publisher