Add melt support when query-planning is enabled (#11088)

dask · May 2, 2024 · 21f73cc · 21f73cc
1 parent 5b0f20e
commit 21f73cc
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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
 ~~~~~~~~~~~~~~~~~~~~~~