From 4b7071f15031cff1af91b831bf82bceb12011ed5 Mon Sep 17 00:00:00 2001 From: rjzamora Date: Thu, 2 May 2024 07:11:07 -0700 Subject: [PATCH] add melt support for dask-expr API --- dask/dataframe/__init__.py | 2 +- dask/dataframe/tests/test_multi.py | 1 - 2 files changed, 1 insertion(+), 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", [