[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

narwhals/_sql/expr.py

@@ -3,31 +3,34 @@
 from typing import TYPE_CHECKING, Any, Callable, Literal, Protocol, cast
 
 from narwhals._compliant.expr import LazyExpr
-from narwhals._compliant.typing import (
-    AliasNames,
-    EvalNames,
-    EvalSeries,
-    WindowFunction,
-)
+from narwhals._compliant.typing import AliasNames, WindowFunction
 from narwhals._compliant.window import WindowInputs
 from narwhals._expression_parsing import (
     combine_alias_output_names,
     combine_evaluate_output_names,
 )
-from narwhals._sql.typing import SQLLazyFrameT, NativeSQLExprT
+from narwhals._sql.typing import NativeSQLExprT, SQLLazyFrameT
 from narwhals._utils import Implementation, Version, not_implemented
 
 if TYPE_CHECKING:
     from collections.abc import Iterable, Sequence
 
     from typing_extensions import Self, TypeIs
 
-    from narwhals._compliant.typing import AliasNames, WindowFunction
+    from narwhals._compliant.typing import (
+        AliasNames,
+        EvalNames,
+        EvalSeries,
+        WindowFunction,
+    )
     from narwhals._expression_parsing import ExprMetadata
     from narwhals._sql.namespace import SQLNamespace
     from narwhals.typing import NumericLiteral, PythonLiteral, RankMethod, TemporalLiteral
 
-class SQLExpr(LazyExpr[SQLLazyFrameT, NativeSQLExprT], Protocol[SQLLazyFrameT, NativeSQLExprT]):
+
+class SQLExpr(
+    LazyExpr[SQLLazyFrameT, NativeSQLExprT], Protocol[SQLLazyFrameT, NativeSQLExprT]
+):
     _call: EvalSeries[SQLLazyFrameT, NativeSQLExprT]
     _evaluate_output_names: EvalNames[SQLLazyFrameT]
     _alias_output_names: AliasNames | None
@@ -173,7 +176,9 @@ def default_window_func(
 
         return self._window_function or default_window_func
 
-    def _function(self, name: str, *args: NativeSQLExprT | PythonLiteral) -> NativeSQLExprT:
+    def _function(
+        self, name: str, *args: NativeSQLExprT | PythonLiteral
+    ) -> NativeSQLExprT:
         return self.__narwhals_namespace__()._function(name, *args)
 
     def _lit(self, value: Any) -> NativeSQLExprT:
@@ -273,7 +278,7 @@ def func(
             }
             return [
                 self._when(
-                    self._window_expression(  
+                    self._window_expression(
                         self._function("count", expr), **window_kwargs
                     )
                     >= self._lit(min_samples),
@@ -494,13 +499,12 @@ def round(self, decimals: int) -> Self:
         return self._with_elementwise(
             lambda expr: self._function("round", expr, self._lit(decimals))
         )
-    # WIP: trying new NativeSQLExprT 
+
+    # WIP: trying new NativeSQLExprT
     def sqrt(self) -> Self:
         def _sqrt(expr: NativeSQLExprT) -> NativeSQLExprT:
             return self._when(
-                expr < self._lit(0),  
-                self._lit(float("nan")),
-                self._function("sqrt", expr),
+                expr < self._lit(0), self._lit(float("nan")), self._function("sqrt", expr)
             )
 
         return self._with_elementwise(_sqrt)
@@ -511,12 +515,12 @@ def exp(self) -> Self:
     def log(self, base: float) -> Self:
         def _log(expr: NativeSQLExprT) -> NativeSQLExprT:
             return self._when(
-                expr < self._lit(0),  
+                expr < self._lit(0),
                 self._lit(float("nan")),
                 self._when(
                     cast("NativeSQLExprT", expr == self._lit(0)),
                     self._lit(float("-inf")),
-                    self._function("log", expr) / self._function("log", self._lit(base)),  
+                    self._function("log", expr) / self._function("log", self._lit(base)),
                 ),
             )
 
@@ -664,19 +668,19 @@ def _rank(
             count_window_kwargs: dict[str, Any] = {"partition_by": (*partition_by, expr)}
             if method == "max":
                 rank_expr = (
-                    self._window_expression(func, **window_kwargs)  
+                    self._window_expression(func, **window_kwargs)
                     + self._window_expression(count_expr, **count_window_kwargs)
                     - self._lit(1)
                 )
             elif method == "average":
                 rank_expr = self._window_expression(func, **window_kwargs) + (
-                    self._window_expression(count_expr, **count_window_kwargs)  
+                    self._window_expression(count_expr, **count_window_kwargs)
                     - self._lit(1)
                 ) / self._lit(2.0)
             else:
                 rank_expr = self._window_expression(func, **window_kwargs)
-                # TODO: @mp, thought I added this to NativeSQLExprT but not working? 
-            return self._when(~self._function("isnull", expr), rank_expr)  # type: ignore[operator] 
+                # TODO: @mp, thought I added this to NativeSQLExprT but not working?
+            return self._when(~self._function("isnull", expr), rank_expr)  # type: ignore[operator]
 
         def _unpartitioned_rank(expr: NativeSQLExprT) -> NativeSQLExprT:
             return _rank(expr, descending=[descending], nulls_last=[True])

narwhals/_sql/typing.py

@@ -1,27 +1,28 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, TypeVar, Protocol
+from typing import TYPE_CHECKING, Any, Protocol, TypeVar
 
 from narwhals._compliant.expr import NativeExpr
 
 if TYPE_CHECKING:
+    from typing_extensions import Self
+
     from narwhals._sql.dataframe import SQLLazyFrame
     from narwhals._sql.expr import SQLExpr
-    from narwhals.dtypes import Boolean
-    from typing_extensions import Self
 
-    # TODO: check we 
+    # TODO: check we
     SQLExprAny = SQLExpr[Any, Any]
     SQLLazyFrameAny = SQLLazyFrame[Any, Any, Any]
 
 SQLExprT = TypeVar("SQLExprT", bound="SQLExprAny")
 SQLExprT_contra = TypeVar("SQLExprT_contra", bound="SQLExprAny", contravariant=True)
 SQLLazyFrameT = TypeVar("SQLLazyFrameT", bound="SQLLazyFrameAny")
 # TODO: @mp, should this be contravariant as to do with function arguments? think through!
-NativeSQLExprT = TypeVar("NativeSQLExprT", bound="NativeSQLExpr") 
+NativeSQLExprT = TypeVar("NativeSQLExprT", bound="NativeSQLExpr")
+
 
 class NativeSQLExpr(NativeExpr, Protocol):
-    # both Self because we're comparing an expression with an expression? 
+    # both Self because we're comparing an expression with an expression?
     def __gt__(self, value: Any, /) -> Self: ...
 
     def __lt__(self, value: Any, /) -> Self: ...
@@ -45,6 +46,3 @@ def __truediv__(self, value: Any, /) -> Self: ...
     # def __mul__(self, value: Self) -> Self: ...
 
     # def __invert__(self, value: Self) -> Self: ...
-
-
-