chore: bigquery signatures, doctests, docs, unused imports

docs/_quarto.yml

@@ -298,8 +298,6 @@ quartodoc:
               package: ibis.expr.types.generic
             - name: Column
               package: ibis.expr.types.generic
-            - name: Deferred
-              package: ibis.common.deferred
             - name: Scalar
               package: ibis.expr.types.generic
 

ibis/backends/sql/compilers/base.py

@@ -11,7 +11,7 @@
 
 import sqlglot as sg
 import sqlglot.expressions as sge
-from koerce import Replace
+from koerce import Replace  # noqa: TCH002
 from public import public
 
 import ibis.common.exceptions as com
@@ -431,7 +431,7 @@ class SQLGlotCompiler(abc.ABC):
     # Constructed dynamically in `__init_subclass__` from their respective
     # UPPERCASE values to handle inheritance, do not modify directly here.
     extra_supported_ops: ClassVar[frozenset[type[ops.Node]]] = frozenset()
-    lowered_ops: ClassVar[dict[type[ops.Node], pats.Replace]] = {}
+    lowered_ops: ClassVar[dict[type[ops.Node], Replace]] = {}
 
     def __init__(self) -> None:
         self.f = FuncGen(copy=self.__class__.copy_func_args)

ibis/backends/sql/compilers/bigquery/__init__.py

@@ -280,7 +280,7 @@ def _compile_python_udf(self, udf_node: ops.ScalarUDF) -> sge.Create:
         signature = [
             sge.ColumnDef(
                 this=sg.to_identifier(name, quoted=self.quoted),
-                kind=type_mapper.from_ibis(param.annotation.pattern.dtype),
+                kind=type_mapper.from_ibis(dt.dtype(param.typehint)),
             )
             for name, param in udf_node.__signature__.parameters.items()
         ]

ibis/common/collections.py

@@ -2,7 +2,6 @@
 
 import collections.abc
 from abc import abstractmethod
-from itertools import tee
 from typing import TYPE_CHECKING, Any, Generic, TypeVar
 
 from koerce import AbstractMeta

ibis/common/deferred.py

@@ -0,0 +1,12 @@
+from __future__ import annotations
+
+from koerce import Deferred, Var
+
+
+class _Variable(Var):
+    def __repr__(self):
+        return self.name
+
+
+# reserved variable name for the value being matched
+_ = Deferred(_Variable("_"))

ibis/common/graph.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 import itertools
-from abc import abstractmethod
 from collections import deque
 from collections.abc import Callable, Iterable, Iterator, KeysView, Mapping, Sequence
 from typing import TYPE_CHECKING, Any, Optional, TypeVar, Union
@@ -47,7 +46,7 @@ def _flatten_collections(node: Any) -> Iterator[N]:
     >>> from ibis.common.grounds import Concrete
     >>> from ibis.common.graph import Node
     >>>
-    >>> class MyNode(Concrete, Node):
+    >>> class MyNode(Node):
     ...     number: int
     ...     string: str
     ...     children: tuple[Node, ...]
@@ -95,7 +94,7 @@ def _recursive_lookup(obj: Any, dct: dict) -> Any:
     >>> from ibis.common.grounds import Concrete
     >>> from ibis.common.graph import Node
     >>>
-    >>> class MyNode(Concrete, Node):
+    >>> class MyNode(Node):
     ...     number: int
     ...     string: str
     ...     children: tuple[Node, ...]

ibis/common/grounds.py

@@ -1,7 +1,5 @@
 from __future__ import annotations
 
-from abc import abstractmethod
-
 from koerce import Annotable, MatchError
 from typing_extensions import Self