simplify dask imports

narwhals-dev · Jan 20, 2025 · 8637a5c · 8637a5c
1 parent d5619d8
commit 8637a5c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/narwhals/_dask/dataframe.py b/narwhals/_dask/dataframe.py
@@ -8,6 +8,7 @@
 from typing import Sequence
 
 import dask.dataframe as dd
+import pandas as pd
 
 from narwhals._dask.utils import add_row_index
 from narwhals._dask.utils import parse_exprs_and_named_exprs
@@ -81,8 +82,6 @@ def with_columns(self, *exprs: DaskExpr, **named_exprs: DaskExpr) -> Self:
         return self._from_native_frame(df)
 
     def collect(self) -> Any:
-        import pandas as pd
-
         from narwhals._pandas_like.dataframe import PandasLikeDataFrame
 
         result = self._native_frame.compute()
@@ -127,8 +126,6 @@ def select(
 
         if not new_series:
             # return empty dataframe, like Polars does
-            import pandas as pd
-
             return self._from_native_frame(
                 dd.from_pandas(pd.DataFrame(), npartitions=self._native_frame.npartitions)
             )

diff --git a/narwhals/_dask/namespace.py b/narwhals/_dask/namespace.py
@@ -99,8 +99,6 @@ def func(df: DaskLazyFrame) -> list[dx.Series]:
         )
 
     def len(self) -> DaskExpr:
-        import pandas as pd
-
         def func(df: DaskLazyFrame) -> list[dx.Series]:
             if not df.columns:
                 return [