chore(typing): resolve mypy strict errors

Following #2077

Author: dangotbanned

narwhals/_arrow/dataframe.py

@@ -618,7 +618,7 @@ def collect(
         self: Self,
         backend: Implementation | None,
         **kwargs: Any,
-    ) -> CompliantDataFrame:
+    ) -> CompliantDataFrame[Any]:
         if backend is Implementation.PYARROW or backend is None:
             from narwhals._arrow.dataframe import ArrowDataFrame
 

narwhals/_expression_parsing.py

@@ -90,7 +90,7 @@ def maybe_evaluate_expr(
 
 
 @overload
-def maybe_evaluate_expr(df: CompliantDataFrame, expr: T) -> T: ...
+def maybe_evaluate_expr(df: CompliantDataFrame[Any], expr: T) -> T: ...
 
 
 def maybe_evaluate_expr(
@@ -152,7 +152,7 @@ def reuse_series_implementation(
     """
     plx = expr.__narwhals_namespace__()
 
-    def func(df: CompliantDataFrame) -> Sequence[CompliantSeries]:
+    def func(df: CompliantDataFrame[Any]) -> Sequence[CompliantSeries]:
         _kwargs = {
             **(call_kwargs or {}),
             **{
@@ -303,7 +303,7 @@ def extract_compliant(
 
 def evaluate_output_names_and_aliases(
     expr: CompliantExpr[Any, Any],
-    df: CompliantDataFrame | CompliantLazyFrame,
+    df: CompliantDataFrame[Any] | CompliantLazyFrame,
     exclude: Sequence[str],
 ) -> tuple[Sequence[str], Sequence[str]]:
     output_names = expr._evaluate_output_names(df)

narwhals/_selectors.py

@@ -5,6 +5,7 @@
 import re
 from functools import partial
 from typing import TYPE_CHECKING
+from typing import Any
 from typing import Callable
 from typing import Collection
 from typing import Generic
@@ -41,7 +42,7 @@
 
 
 SeriesT = TypeVar("SeriesT", bound="CompliantSeries")
-FrameT = TypeVar("FrameT", bound="CompliantDataFrame | CompliantLazyFrame")
+FrameT = TypeVar("FrameT", bound="CompliantDataFrame[Any] | CompliantLazyFrame")
 SelectorOrExpr: TypeAlias = (
     "CompliantSelector[FrameT, SeriesT] | CompliantExpr[FrameT, SeriesT]"
 )
@@ -265,14 +266,16 @@ def names(df: FrameT) -> Sequence[str]:
             return self._to_expr() & other
 
     def __invert__(self: Self) -> CompliantSelector[FrameT, SeriesT]:
-        return self.selectors.all() - self
+        return self.selectors.all() - self  # type: ignore[no-any-return]
 
     def __repr__(self: Self) -> str:  # pragma: no cover
         s = f"depth={self._depth}, " if is_tracks_depth(self._implementation) else ""
         return f"{type(self).__name__}({s}function_name={self._function_name})"
 
 
 def _eval_lhs_rhs(
-    df: CompliantDataFrame | CompliantLazyFrame, lhs: CompliantExpr, rhs: CompliantExpr
+    df: CompliantDataFrame[Any] | CompliantLazyFrame,
+    lhs: CompliantExpr[Any, Any],
+    rhs: CompliantExpr[Any, Any],
 ) -> tuple[Sequence[str], Sequence[str]]:
     return lhs._evaluate_output_names(df), rhs._evaluate_output_names(df)

narwhals/_spark_like/dataframe.py

@@ -163,7 +163,7 @@ def collect(
         self: Self,
         backend: ModuleType | Implementation | str | None,
         **kwargs: Any,
-    ) -> CompliantDataFrame:
+    ) -> CompliantDataFrame[Any]:
         if backend is Implementation.PANDAS:
             import pandas as pd  # ignore-banned-import
 

narwhals/typing.py

@@ -93,7 +93,7 @@ def schema(self) -> Mapping[str, DType]: ...
 
 
 CompliantFrameT = TypeVar(
-    "CompliantFrameT", bound="CompliantDataFrame | CompliantLazyFrame"
+    "CompliantFrameT", bound="CompliantDataFrame[Any] | CompliantLazyFrame"
 )
 
 

narwhals/utils.py

@@ -1309,7 +1309,7 @@ def dtype_matches_time_unit_and_time_zone(
 
 
 def get_column_names(
-    df: NativeFrame | CompliantDataFrame | CompliantLazyFrame,
+    df: NativeFrame | CompliantDataFrame[Any] | CompliantLazyFrame,
 ) -> Sequence[str]:
     return df.columns
 
@@ -1319,7 +1319,9 @@ def _hasattr_static(obj: Any, attr: str) -> bool:
     return getattr_static(obj, attr, sentinel) is not sentinel
 
 
-def is_compliant_dataframe(obj: Any) -> TypeIs[CompliantDataFrame]:
+def is_compliant_dataframe(
+    obj: Any | CompliantDataFrame[CompliantSeriesT_co],
+) -> TypeIs[CompliantDataFrame[CompliantSeriesT_co]]:
     return _hasattr_static(obj, "__narwhals_dataframe__")