add melt support for dask-expr API

rjzamora · May 2, 2024 · 4b7071f · 4b7071f
1 parent 814ed3b
commit 4b7071f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition 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",
     [
-Original file line number
+Diff line change
@@ Expand Up / @@ -1781,7 +1781,6 @@ def fix_index(out, dtype): @@
                 )
-    @pytest.mark.xfail(DASK_EXPR_ENABLED, reason="melt not supported yet")
     @pytest.mark.parametrize(
         "kwargs",
         [
@@ Expand Down @@