blob: d72da05c35e6566e15ebe47675f0f0f239e36ef5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
from __future__ import annotations
from typing import TYPE_CHECKING
from narwhals._arrow.expr import ArrowExpr
from narwhals._compliant import CompliantSelector, EagerSelectorNamespace
if TYPE_CHECKING:
from narwhals._arrow.dataframe import ArrowDataFrame # noqa: F401
from narwhals._arrow.series import ArrowSeries # noqa: F401
class ArrowSelectorNamespace(EagerSelectorNamespace["ArrowDataFrame", "ArrowSeries"]):
@property
def _selector(self) -> type[ArrowSelector]:
return ArrowSelector
class ArrowSelector(CompliantSelector["ArrowDataFrame", "ArrowSeries"], ArrowExpr): # type: ignore[misc]
def _to_expr(self) -> ArrowExpr:
return ArrowExpr(
self._call,
depth=self._depth,
function_name=self._function_name,
evaluate_output_names=self._evaluate_output_names,
alias_output_names=self._alias_output_names,
backend_version=self._backend_version,
version=self._version,
)
|
