From a7a963a6784c3fae329cf6e4dfbb5a4d73077b84 Mon Sep 17 00:00:00 2001 From: Patrick Hoefler <61934744+phofl@users.noreply.github.com> Date: Fri, 15 Nov 2024 20:11:27 +0100 Subject: [PATCH] Remove ``from_dask_dataframe`` (#1167) --- dask_expr/_collection.py | 17 ----------------- dask_expr/io/tests/test_io.py | 6 ------ 2 files changed, 23 deletions(-) diff --git a/dask_expr/_collection.py b/dask_expr/_collection.py index 4d38808f..40afc6a8 100644 --- a/dask_expr/_collection.py +++ b/dask_expr/_collection.py @@ -5020,23 +5020,6 @@ def from_dict( ) -def from_dask_dataframe(*args, **kwargs) -> FrameBase: - """Create a dask-expr collection from a legacy dask-dataframe collection - - WARNING: This API is deprecated. Please use `from_legacy_dataframe`. - - Parameters - ---------- - optimize - Whether to optimize the graph before conversion. - """ - warnings.warn( - "`from_dask_dataframe` is deprecated, please use `from_legacy_dataframe`.", - FutureWarning, - ) - return from_legacy_dataframe(*args, **kwargs) - - def from_legacy_dataframe(ddf: _Frame, optimize: bool = True) -> FrameBase: """Create a dask-expr collection from a legacy dask-dataframe collection diff --git a/dask_expr/io/tests/test_io.py b/dask_expr/io/tests/test_io.py index e6ec76d3..ca284a9e 100644 --- a/dask_expr/io/tests/test_io.py +++ b/dask_expr/io/tests/test_io.py @@ -14,7 +14,6 @@ DataFrame, from_array, from_dask_array, - from_dask_dataframe, from_dict, from_legacy_dataframe, from_map, @@ -235,11 +234,6 @@ def test_from_legacy_dataframe(optimize): assert isinstance(df.expr, Expr) assert_eq(df, ddf) - # Check deprecated API - with pytest.warns(FutureWarning, match="deprecated"): - df2 = from_dask_dataframe(ddf, optimize=optimize) - assert_eq(df, df2) - @pytest.mark.parametrize("optimize", [True, False]) def test_to_legacy_dataframe(optimize):