fixes

Author: TrevorBergeron

packages/bigframes/bigframes/session/bq_caching_executor.py

@@ -129,7 +129,7 @@ def to_sql(
             else array_value.node
         )
         node = self._substitute_large_local_sources(node)
-        compiled = compile.compiler().compile_sql(
+        compiled = compile.compile_sql(
             compile.CompileRequest(node, sort_rows=ordered),
             compiler_name=self._compiler_name,
         )
@@ -256,7 +256,7 @@ def _export_gbq(
         # validate destination table
         existing_table = self._maybe_find_existing_table(spec)
 
-        compiled = compile.compiler().compile_sql(
+        compiled = compile.compile_sql(
             compile.CompileRequest(plan, sort_rows=False),
             compiler_name=self._compiler_name,
         )

packages/bigframes/tests/unit/session/test_proxy_executor.py

@@ -65,7 +65,7 @@ def test_execute_experimental_routes_to_sqlglot(mock_executor, monkeypatch):
     monkeypatch.setattr(bigframes.options.experiments, "sql_compiler", "experimental")
     mock_executor.execute(array_value, execution_spec)
 
-    execution_spec.add_labels.assert_called_once_with(("bigframes-compiler", "sqlglot"))
+    execution_spec.add_labels.assert_called_once_with({"bigframes-compiler": "sqlglot"})
     mock_executor._sqlglot_executor.execute.assert_called_once_with(
         array_value, execution_spec
     )
@@ -86,7 +86,7 @@ def test_execute_stable_routes_to_sqlglot_success(mock_executor, monkeypatch):
         mock_executor.execute(array_value, execution_spec)
 
     execution_spec.add_labels.assert_called_once_with(
-        ("bigframes-compiler", "sqlglot-123456789012")
+        {"bigframes-compiler": "sqlglot-123456789012"}
     )
     mock_executor._sqlglot_executor.execute.assert_called_once_with(
         array_value, execution_spec
@@ -117,8 +117,8 @@ def test_execute_stable_routes_to_sqlglot_fallback_to_ibis(mock_executor, monkey
 
     execution_spec.add_labels.assert_has_calls(
         [
-            mock.call(("bigframes-compiler", "sqlglot-123456789012")),
-            mock.call(("bigframes-compiler", "ibis-123456789012")),
+            mock.call({"bigframes-compiler": "sqlglot-123456789012"}),
+            mock.call({"bigframes-compiler": "ibis-123456789012"}),
         ]
     )