feat(pandas): support Table.sample

ibis-project · Oct 17, 2023 · 77215be · 77215be
1 parent f3d044c
commit 77215be
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/ibis/backends/pandas/execution/generic.py b/ibis/backends/pandas/execution/generic.py
@@ -1450,3 +1450,8 @@ def execute_table_array_view(op, _, **kwargs):
 @execute_node.register(ops.InMemoryTable)
 def execute_in_memory_table(op, **kwargs):
     return op.data.to_frame()
+
+
+@execute_node.register(ops.Sample, pd.DataFrame, object, object)
+def execute_sample(op, data, fraction, seed, **kwargs):
+    return data.sample(frac=fraction, random_state=seed)
diff --git a/ibis/backends/tests/test_generic.py b/ibis/backends/tests/test_generic.py
@@ -1537,7 +1537,6 @@ def test_dynamic_table_slice_with_computed_offset(backend):
         "druid",
         "flink",
         "impala",
-        "pandas",
         "polars",
         "pyspark",
         "snowflake",
@@ -1566,7 +1565,6 @@ def test_sample(backend):
         "druid",
         "flink",
         "impala",
-        "pandas",
         "polars",
         "pyspark",
         "snowflake",
@@ -1591,7 +1589,6 @@ def test_sample_memtable(con, backend):
         "mssql",
         "mysql",
         "oracle",
-        "pandas",
         "polars",
         "postgres",
         "pyspark",