From 8637a5cda0ba4a2cf119a842f9cc9eb05f6cc087 Mon Sep 17 00:00:00 2001 From: Marco Gorelli <33491632+MarcoGorelli@users.noreply.github.com> Date: Mon, 20 Jan 2025 13:58:38 +0400 Subject: [PATCH] simplify dask imports --- narwhals/_dask/dataframe.py | 5 +---- narwhals/_dask/namespace.py | 2 -- 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/narwhals/_dask/dataframe.py b/narwhals/_dask/dataframe.py index de169fd5c..d493a9631 100644 --- 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 index b5fcd82f7..626937af3 100644 --- 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 [