style: Somewhat alphabetical member order

Author: dangotbanned

narwhals/_plan/compliant/column.py

@@ -25,14 +25,16 @@
 class SupportsBroadcast(Protocol[SeriesT, LengthT]):
     """Minimal broadcasting for `Expr` results."""
 
-    @classmethod
-    def from_series(cls, series: SeriesT, /) -> Self: ...
-    def to_series(self) -> SeriesT: ...
-    def broadcast(self, length: LengthT, /) -> SeriesT: ...
     def _length(self) -> LengthT:
         """Return the length of the current expression."""
         ...
 
+    @classmethod
+    def _length_all(
+        cls, exprs: Sequence[SupportsBroadcast[SeriesT, LengthT]], /
+    ) -> Sequence[LengthT]:
+        return [e._length() for e in exprs]
+
     @classmethod
     def _length_max(cls, lengths: Sequence[LengthT], /) -> LengthT:
         """Return the maximum length among `exprs`."""
@@ -44,12 +46,6 @@ def _length_required(
     ) -> LengthT | None:
         """Return the broadcast length, if all lengths do not equal the maximum."""
 
-    @classmethod
-    def _length_all(
-        cls, exprs: Sequence[SupportsBroadcast[SeriesT, LengthT]], /
-    ) -> Sequence[LengthT]:
-        return [e._length() for e in exprs]
-
     @classmethod
     def align(
         cls, *exprs: OneOrIterable[SupportsBroadcast[SeriesT, LengthT]]
@@ -63,6 +59,11 @@ def align(
             for e in exprs:
                 yield e.broadcast(length)
 
+    def broadcast(self, length: LengthT, /) -> SeriesT: ...
+    @classmethod
+    def from_series(cls, series: SeriesT, /) -> Self: ...
+    def to_series(self) -> SeriesT: ...
+
 
 class EagerBroadcast(Sized, SupportsBroadcast[SeriesT, int], Protocol[SeriesT]):
     """Determines expression length via the size of the container."""
@@ -85,6 +86,8 @@ def _length_required(
 
 
 class ExprDispatch(StoresVersion, Protocol[FrameT_contra, R_co, NamespaceT_co]):
+    # NOTE: Needs to stay `covariant` and never be used as a parameter
+    def __narwhals_namespace__(self) -> NamespaceT_co: ...
     @classmethod
     def from_ir(cls, node: ir.ExprIR, frame: FrameT_contra, name: str) -> R_co:
         obj = cls.__new__(cls)
@@ -94,6 +97,3 @@ def from_ir(cls, node: ir.ExprIR, frame: FrameT_contra, name: str) -> R_co:
     @classmethod
     def from_named_ir(cls, named_ir: ir.NamedIR[ir.ExprIR], frame: FrameT_contra) -> R_co:
         return cls.from_ir(named_ir.expr, frame, named_ir.name)
-
-    # NOTE: Needs to stay `covariant` and never be used as a parameter
-    def __narwhals_namespace__(self) -> NamespaceT_co: ...

narwhals/_plan/compliant/dataframe.py

@@ -38,42 +38,43 @@ class CompliantFrame(StoresVersion, Protocol[ColumnT_co, NativeFrameT]):
     _native: NativeFrameT
 
     def __narwhals_namespace__(self) -> Any: ...
+    def _evaluate_irs(
+        self, nodes: Iterable[NamedIR[ir.ExprIR]], /
+    ) -> Iterator[ColumnT_co]: ...
     @property
     def _group_by(self) -> type[CompliantGroupBy[Self]]: ...
-    @property
-    def native(self) -> NativeFrameT:
-        return self._native
+    def _with_native(self, native: NativeFrameT) -> Self:
+        return self.from_native(native, self.version)
 
-    @property
-    def columns(self) -> list[str]: ...
-    def to_narwhals(self) -> BaseFrame[NativeFrameT]: ...
     @classmethod
     def from_native(cls, native: NativeFrameT, /, version: Version) -> Self:
         obj = cls.__new__(cls)
         obj._native = native
         obj._version = version
         return obj
 
-    def _with_native(self, native: NativeFrameT) -> Self:
-        return self.from_native(native, self.version)
+    @property
+    def native(self) -> NativeFrameT:
+        return self._native
 
+    def to_narwhals(self) -> BaseFrame[NativeFrameT]: ...
+    @property
+    def columns(self) -> list[str]: ...
+    def drop(self, columns: Sequence[str], *, strict: bool = True) -> Self: ...
+    def drop_nulls(self, subset: Sequence[str] | None) -> Self: ...
     @property
     def schema(self) -> Mapping[str, DType]: ...
-    def _evaluate_irs(
-        self, nodes: Iterable[NamedIR[ir.ExprIR]], /
-    ) -> Iterator[ColumnT_co]: ...
     def select(self, irs: Seq[NamedIR]) -> Self: ...
     def select_names(self, *column_names: str) -> Self: ...
-    def with_columns(self, irs: Seq[NamedIR]) -> Self: ...
     def sort(self, by: Seq[NamedIR], options: SortMultipleOptions) -> Self: ...
-    def drop(self, columns: Sequence[str], *, strict: bool = True) -> Self: ...
-    def drop_nulls(self, subset: Sequence[str] | None) -> Self: ...
+    def with_columns(self, irs: Seq[NamedIR]) -> Self: ...
 
 
 class CompliantDataFrame(
     CompliantFrame[SeriesT, NativeDataFrameT],
     Protocol[SeriesT, NativeDataFrameT, NativeSeriesT],
 ):
+    def __len__(self) -> int: ...
     @property
     def _group_by(self) -> type[DataFrameGroupBy[Self]]: ...
     @property
@@ -101,7 +102,7 @@ def group_by_resolver(self, resolver: GroupByResolver, /) -> DataFrameGroupBy[Se
         """
         return self._group_by.from_resolver(self, resolver)
 
-    def to_narwhals(self) -> DataFrame[NativeDataFrameT, NativeSeriesT]: ...
+    def row(self, index: int) -> tuple[Any, ...]: ...
     @overload
     def to_dict(self, *, as_series: Literal[True]) -> dict[str, SeriesT]: ...
     @overload
@@ -113,18 +114,17 @@ def to_dict(
     def to_dict(
         self, *, as_series: bool
     ) -> dict[str, SeriesT] | dict[str, list[Any]]: ...
-    def __len__(self) -> int: ...
+    def to_narwhals(self) -> DataFrame[NativeDataFrameT, NativeSeriesT]: ...
     def with_row_index(self, name: str) -> Self: ...
-    def row(self, index: int) -> tuple[Any, ...]: ...
 
 
 class EagerDataFrame(
     CompliantDataFrame[SeriesT, NativeDataFrameT, NativeSeriesT],
     Protocol[SeriesT, NativeDataFrameT, NativeSeriesT],
 ):
+    def __narwhals_namespace__(self) -> EagerNamespace[Self, SeriesT, Any, Any]: ...
     @property
     def _group_by(self) -> type[EagerDataFrameGroupBy[Self]]: ...
-    def __narwhals_namespace__(self) -> EagerNamespace[Self, SeriesT, Any, Any]: ...
     def select(self, irs: Seq[NamedIR]) -> Self:
         return self.__narwhals_namespace__()._concat_horizontal(self._evaluate_irs(irs))
 

narwhals/_plan/compliant/expr.py

@@ -33,20 +33,19 @@ class CompliantExpr(StoresVersion, Protocol[FrameT_contra, SeriesT_co]):
     _evaluated: Any
     """Compliant or native value."""
 
-    @property
-    def name(self) -> str: ...
+    def _with_native(self, native: Any, name: str, /) -> Self:
+        return self.from_native(native, name or self.name, self.version)
+
     @classmethod
     def from_native(
         cls, native: Any, name: str = "", /, version: Version = Version.MAIN
     ) -> Self: ...
-    def _with_native(self, native: Any, name: str, /) -> Self:
-        return self.from_native(native, name or self.name, self.version)
-
+    @property
+    def name(self) -> str: ...
     # series & scalar
     def abs(self, node: FunctionExpr[F.Abs], frame: FrameT_contra, name: str) -> Self: ...
+    def binary_expr(self, node: BinaryExpr, frame: FrameT_contra, name: str) -> Self: ...
     def cast(self, node: ir.Cast, frame: FrameT_contra, name: str) -> Self: ...
-    def pow(self, node: FunctionExpr[F.Pow], frame: FrameT_contra, name: str) -> Self: ...
-    def not_(self, node: FunctionExpr[Not], frame: FrameT_contra, name: str) -> Self: ...
     def fill_null(
         self, node: FunctionExpr[F.FillNull], frame: FrameT_contra, name: str
     ) -> Self: ...
@@ -62,57 +61,49 @@ def is_nan(
     def is_null(
         self, node: FunctionExpr[IsNull], frame: FrameT_contra, name: str
     ) -> Self: ...
-    def binary_expr(self, node: BinaryExpr, frame: FrameT_contra, name: str) -> Self: ...
-    def ternary_expr(
-        self, node: ir.TernaryExpr, frame: FrameT_contra, name: str
+    def map_batches(
+        self, node: ir.AnonymousExpr, frame: FrameT_contra, name: str
     ) -> Self: ...
+    def not_(self, node: FunctionExpr[Not], frame: FrameT_contra, name: str) -> Self: ...
     def over(self, node: ir.WindowExpr, frame: FrameT_contra, name: str) -> Self: ...
     # NOTE: `Scalar` is returned **only** for un-partitioned `OrderableAggExpr`
     # e.g. `nw.col("a").first().over(order_by="b")`
     def over_ordered(
         self, node: ir.OrderedWindowExpr, frame: FrameT_contra, name: str
     ) -> Self | CompliantScalar[FrameT_contra, SeriesT_co]: ...
-    def map_batches(
-        self, node: ir.AnonymousExpr, frame: FrameT_contra, name: str
-    ) -> Self: ...
+    def pow(self, node: FunctionExpr[F.Pow], frame: FrameT_contra, name: str) -> Self: ...
     def rolling_expr(
         self, node: ir.RollingExpr, frame: FrameT_contra, name: str
     ) -> Self: ...
-    # series only (section 3)
+    def ternary_expr(
+        self, node: ir.TernaryExpr, frame: FrameT_contra, name: str
+    ) -> Self: ...
+    # series only
+    def filter(self, node: ir.Filter, frame: FrameT_contra, name: str) -> Self: ...
     def sort(self, node: ir.Sort, frame: FrameT_contra, name: str) -> Self: ...
     def sort_by(self, node: ir.SortBy, frame: FrameT_contra, name: str) -> Self: ...
-    def filter(self, node: ir.Filter, frame: FrameT_contra, name: str) -> Self: ...
     # series -> scalar
-    def first(
-        self, node: agg.First, frame: FrameT_contra, name: str
-    ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
-    def last(
-        self, node: agg.Last, frame: FrameT_contra, name: str
+    def all(
+        self, node: FunctionExpr[boolean.All], frame: FrameT_contra, name: str
     ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
-    def arg_min(
-        self, node: agg.ArgMin, frame: FrameT_contra, name: str
+    def any(
+        self, node: FunctionExpr[boolean.Any], frame: FrameT_contra, name: str
     ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
     def arg_max(
         self, node: agg.ArgMax, frame: FrameT_contra, name: str
     ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
-    def sum(
-        self, node: agg.Sum, frame: FrameT_contra, name: str
-    ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
-    def n_unique(
-        self, node: agg.NUnique, frame: FrameT_contra, name: str
-    ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
-    def std(
-        self, node: agg.Std, frame: FrameT_contra, name: str
-    ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
-    def var(
-        self, node: agg.Var, frame: FrameT_contra, name: str
-    ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
-    def quantile(
-        self, node: agg.Quantile, frame: FrameT_contra, name: str
+    def arg_min(
+        self, node: agg.ArgMin, frame: FrameT_contra, name: str
     ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
     def count(
         self, node: agg.Count, frame: FrameT_contra, name: str
     ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
+    def first(
+        self, node: agg.First, frame: FrameT_contra, name: str
+    ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
+    def last(
+        self, node: agg.Last, frame: FrameT_contra, name: str
+    ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
     def len(
         self, node: agg.Len, frame: FrameT_contra, name: str
     ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
@@ -128,11 +119,20 @@ def median(
     def min(
         self, node: agg.Min, frame: FrameT_contra, name: str
     ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
-    def all(
-        self, node: FunctionExpr[boolean.All], frame: FrameT_contra, name: str
+    def n_unique(
+        self, node: agg.NUnique, frame: FrameT_contra, name: str
     ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
-    def any(
-        self, node: FunctionExpr[boolean.Any], frame: FrameT_contra, name: str
+    def quantile(
+        self, node: agg.Quantile, frame: FrameT_contra, name: str
+    ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
+    def sum(
+        self, node: agg.Sum, frame: FrameT_contra, name: str
+    ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
+    def std(
+        self, node: agg.Std, frame: FrameT_contra, name: str
+    ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...
+    def var(
+        self, node: agg.Var, frame: FrameT_contra, name: str
     ) -> CompliantScalar[FrameT_contra, SeriesT_co]: ...