Skip to content

Commit 6e4d621

Browse files
anjakefalacpcloud
authored andcommitted
feat(pandas): support memtable in pandas backend
Closes #5467
1 parent 4a957fd commit 6e4d621

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

ibis/backends/pandas/execution/generic.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1442,6 +1442,11 @@ def execute_zero_if_null_series(op, data, **kwargs):
14421442
return data.replace({np.nan: zero, None: zero, pd.NA: zero})
14431443

14441444

1445+
@execute_node.register(ops.InMemoryTable)
1446+
def execute_in_memory_table(op, **kwargs):
1447+
return op.data.to_frame()
1448+
1449+
14451450
@execute_node.register(
14461451
ops.ZeroIfNull,
14471452
(type(None), type(pd.NA), numbers.Real, np.integer, np.floating),

ibis/backends/tests/test_generic.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -916,7 +916,7 @@ def test_literal_na(con, dtype):
916916

917917

918918
@pytest.mark.notimpl(["datafusion"], raises=com.OperationNotDefinedError)
919-
@pytest.mark.notimpl(["dask", "pandas"], raises=com.UnboundExpressionError)
919+
@pytest.mark.notimpl(["dask"], raises=com.UnboundExpressionError)
920920
def test_memtable_bool_column(backend, con, monkeypatch):
921921
monkeypatch.setattr(ibis.options, "default_backend", con)
922922

@@ -925,7 +925,7 @@ def test_memtable_bool_column(backend, con, monkeypatch):
925925

926926

927927
@pytest.mark.notimpl(["datafusion"], raises=com.OperationNotDefinedError)
928-
@pytest.mark.notimpl(["dask", "pandas"], raises=com.UnboundExpressionError)
928+
@pytest.mark.notimpl(["dask"], raises=com.UnboundExpressionError)
929929
@pytest.mark.broken(
930930
["druid"],
931931
raises=AssertionError,

0 commit comments

Comments
 (0)