lint

Author: TrevorBergeron

packages/bigframes/bigframes/session/bq_caching_executor.py

@@ -81,7 +81,7 @@ def __init__(
         enable_polars_execution: bool = False,
         publisher: bigframes.core.events.Publisher,
         labels: Mapping[str, str] = {},
-        compiler_name: str = "ibis",
+        compiler_name: Literal["ibis", "sqlglot"] = "ibis",
         cache: Optional[execution_cache.ExecutionCache] = None,
     ):
         self.bqclient = bqclient

packages/bigframes/bigframes/session/proxy_executor.py

@@ -14,11 +14,12 @@
 
 from __future__ import annotations
 
-from typing import Mapping, Optional
 import typing
 import uuid
 import warnings
+from typing import Mapping, Optional
 
+import google.cloud.bigquery as bigquery
 import google.cloud.exceptions
 
 import bigframes.core