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