Skip to content

Commit fbf54f1

Browse files
committed
Add wrapper for register table provider
1 parent 2764bca commit fbf54f1

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

python/datafusion/context.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -685,6 +685,14 @@ def deregister_table(self, name: str) -> None:
685685
"""Remove a table from the session."""
686686
self.ctx.deregister_table(name)
687687

688+
def register_table_provider(self, name: str, provider: Any) -> None:
689+
"""Register a table provider.
690+
691+
This table provider must have a method called ``__datafusion_table_provider__``
692+
which returns a PyCapsule that exposes a ``FFI_TableProvider``.
693+
"""
694+
self.ctx.register_table_provider(name, provider)
695+
688696
def register_record_batches(
689697
self, name: str, partitions: list[list[pyarrow.RecordBatch]]
690698
) -> None:

0 commit comments

Comments
 (0)