keep going

Author: MarcoGorelli

narwhals/_arrow/group_by.py

@@ -48,10 +48,7 @@ def __init__(
         self._keys = list(keys)
         self._grouped = pa.TableGroupBy(self._df._native_frame, list(self._keys))
 
-    def agg(
-        self: Self,
-        *exprs: ArrowExpr,
-    ) -> ArrowDataFrame:
+    def agg(self: Self, *exprs: ArrowExpr) -> ArrowDataFrame:
         for expr in exprs:
             if expr._output_names is None:
                 msg = "group_by.agg"

narwhals/_duckdb/group_by.py

@@ -9,7 +9,7 @@
     from typing_extensions import Self
 
     from narwhals._duckdb.dataframe import DuckDBLazyFrame
-    from narwhals._duckdb.typing import IntoDuckDBExpr
+    from narwhals._duckdb.expr import DuckDBExpr
 
 
 class DuckDBGroupBy:
@@ -27,7 +27,7 @@ def __init__(
 
     def agg(
         self: Self,
-        *exprs: IntoDuckDBExpr,
+        *exprs: DuckDBExpr,
     ) -> DuckDBLazyFrame:
         output_names: list[str] = copy(self._keys)
         for expr in exprs: