address comments

Author: chelsea-lin

bigframes/core/compile/__init__.py

@@ -13,11 +13,28 @@
 # limitations under the License.
 from __future__ import annotations
 
+from typing import Any
+
+from bigframes import options
 from bigframes.core.compile.api import test_only_ibis_inferred_schema
 from bigframes.core.compile.configs import CompileRequest, CompileResult
 
+
+def compiler() -> Any:
+    """Returns the appropriate compiler module based on session options."""
+    if options.experiments.sql_compiler == "experimental":
+        import bigframes.core.compile.sqlglot.compiler as sqlglot_compiler
+
+        return sqlglot_compiler
+    else:
+        import bigframes.core.compile.ibis_compiler.ibis_compiler as ibis_compiler
+
+        return ibis_compiler
+
+
 __all__ = [
     "test_only_ibis_inferred_schema",
     "CompileRequest",
     "CompileResult",
+    "compiler",
 ]

bigframes/session/bq_caching_executor.py

@@ -31,8 +31,6 @@
 import bigframes.constants
 import bigframes.core
 from bigframes.core import bq_data, compile, local_data, rewrite
-import bigframes.core.compile.ibis_compiler.ibis_compiler as ibis_compiler
-import bigframes.core.compile.sqlglot.compiler as sqlglot_compiler
 import bigframes.core.compile.sqlglot.sqlglot_ir as sqlglot_ir
 import bigframes.core.events
 import bigframes.core.guid
@@ -177,12 +175,9 @@ def to_sql(
             else array_value.node
         )
         node = self._substitute_large_local_sources(node)
-        compiler = (
-            sqlglot_compiler
-            if options.experiments.sql_compiler == "experimental"
-            else ibis_compiler
+        compiled = compile.compiler().compile_sql(
+            compile.CompileRequest(node, sort_rows=ordered)
         )
-        compiled = compiler.compile_sql(compile.CompileRequest(node, sort_rows=ordered))
         return compiled.sql
 
     def execute(
@@ -298,12 +293,9 @@ def _export_gbq(
         # validate destination table
         existing_table = self._maybe_find_existing_table(spec)
 
-        compiler = (
-            sqlglot_compiler
-            if options.experiments.sql_compiler == "experimental"
-            else ibis_compiler
+        compiled = compile.compiler().compile_sql(
+            compile.CompileRequest(plan, sort_rows=False)
         )
-        compiled = compiler.compile_sql(compile.CompileRequest(plan, sort_rows=False))
         sql = compiled.sql
 
         if (existing_table is not None) and _if_schema_match(
@@ -654,12 +646,7 @@ def _execute_plan_gbq(
                 ]
                 cluster_cols = cluster_cols[:_MAX_CLUSTER_COLUMNS]
 
-        compiler = (
-            sqlglot_compiler
-            if options.experiments.sql_compiler == "experimental"
-            else ibis_compiler
-        )
-        compiled = compiler.compile_sql(
+        compiled = compile.compiler().compile_sql(
             compile.CompileRequest(
                 plan,
                 sort_rows=ordered,