fix(examples): use memtables for examples to allow backends that do not support temporary tables to support examples

ibis/backends/datafusion/__init__.py

@@ -9,6 +9,7 @@
 
 import datafusion as df
 import pyarrow as pa
+import pyarrow.dataset as ds
 import pyarrow_hotfix  # noqa: F401
 import sqlglot as sg
 import sqlglot.expressions as sge
@@ -372,7 +373,7 @@ def _register(
             self.con.deregister_table(table_name)
             self.con.register_record_batches(table_name, [[source]])
             return self.table(table_name)
-        elif isinstance(source, pa.dataset.Dataset):
+        elif isinstance(source, ds.Dataset):
             self.con.deregister_table(table_name)
             self.con.register_dataset(table_name, source)
             return self.table(table_name)

ibis/backends/datafusion/tests/test_register.py

@@ -2,7 +2,6 @@
 
 import pandas as pd
 import pyarrow as pa
-import pyarrow.dataset as ds
 import pytest
 
 import ibis
@@ -34,22 +33,24 @@ def test_register_pandas(conn):
     df = pd.DataFrame({"x": [1, 2, 3]})
     with pytest.warns(FutureWarning, match="v9.1"):
         conn.register(df, "my_table")
-        assert conn.table("my_table").x.sum().execute() == 6
+    assert conn.table("my_table").x.sum().execute() == 6
 
 
 def test_register_batches(conn):
     batch = pa.record_batch([pa.array([1, 2, 3])], names=["x"])
     with pytest.warns(FutureWarning, match="v9.1"):
         conn.register(batch, "my_table")
-        assert conn.table("my_table").x.sum().execute() == 6
+    assert conn.table("my_table").x.sum().execute() == 6
 
 
 def test_register_dataset(conn):
+    import pyarrow.dataset as ds
+
     tab = pa.table({"x": [1, 2, 3]})
     dataset = ds.InMemoryDataset(tab)
     with pytest.warns(FutureWarning, match="v9.1"):
         conn.register(dataset, "my_table")
-        assert conn.table("my_table").x.sum().execute() == 6
+    assert conn.table("my_table").x.sum().execute() == 6
 
 
 def test_create_table_with_uppercase_name(conn):

ibis/examples/__init__.py

@@ -78,11 +78,9 @@
                     if pyarrow.types.is_null(field.type):
                         table = table.set_column(i, field.name, table[i].cast("string"))
 
-            # TODO: It should be possible to avoid this memtable call, once all
-            # backends support passing a `pyarrow.Table` to `create_table`
-            # directly.
-            obj = ibis.memtable(table)
-            return backend.create_table(table_name, obj, temp=True, overwrite=True)
+            obj = ibis.memtable(table, name=table_name)
+            backend._register_in_memory_tables(obj)
+            return backend.table(table_name).cache()
 
 
 _FETCH_DOCSTRING_TEMPLATE = """\