From 21f73cc0821336b4972e22794289f5fb9278ceb8 Mon Sep 17 00:00:00 2001 From: "Richard (Rick) Zamora" Date: Thu, 2 May 2024 15:47:17 -0500 Subject: [PATCH] Add ``melt`` support when query-planning is enabled (#11088) --- dask/dataframe/__init__.py | 2 +- dask/dataframe/tests/test_multi.py | 1 - docs/source/dataframe-api.rst | 2 ++ 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/dask/dataframe/__init__.py b/dask/dataframe/__init__.py index 237274f64da..029eef5c54a 100644 --- a/dask/dataframe/__init__.py +++ b/dask/dataframe/__init__.py @@ -132,6 +132,7 @@ def _dask_expr_enabled() -> bool: isna, map_overlap, map_partitions, + melt, merge, merge_asof, pivot_table, @@ -175,7 +176,6 @@ def inner_func(*args, **kwargs): return inner_func _Frame = raise_not_implemented_error("_Frame") # type: ignore - melt = raise_not_implemented_error("melt") # Due to the natural circular imports caused from dask-expr # wanting to import things from dask.dataframe, this module's init diff --git a/dask/dataframe/tests/test_multi.py b/dask/dataframe/tests/test_multi.py index 0f9ce94a0cf..47f05028b92 100644 --- a/dask/dataframe/tests/test_multi.py +++ b/dask/dataframe/tests/test_multi.py @@ -1781,7 +1781,6 @@ def fix_index(out, dtype): ) -@pytest.mark.xfail(DASK_EXPR_ENABLED, reason="melt not supported yet") @pytest.mark.parametrize( "kwargs", [ diff --git a/docs/source/dataframe-api.rst b/docs/source/dataframe-api.rst index d9565068391..0d869ba05d6 100644 --- a/docs/source/dataframe-api.rst +++ b/docs/source/dataframe-api.rst @@ -72,6 +72,7 @@ DataFrame DataFrame.mean DataFrame.median DataFrame.median_approximate + DataFrame.melt DataFrame.memory_usage DataFrame.memory_usage_per_partition DataFrame.merge @@ -549,6 +550,7 @@ Reshape DataFrames get_dummies pivot_table + melt Concatenate DataFrames ~~~~~~~~~~~~~~~~~~~~~~