feat: update register_dataset method to accept any __arrow_c_stream__ source

Author: kosiew

python/datafusion/context.py

@@ -22,8 +22,8 @@
 import warnings
 from typing import TYPE_CHECKING, Any, Protocol
 
-from datafusion.types import ensure_pyarrow_type
 from datafusion.common import DataTypeMap
+from datafusion.types import ensure_pyarrow_type
 
 try:
     from warnings import deprecated  # Python 3.13+
@@ -956,12 +956,16 @@ def register_avro(
             name, str(path), schema, file_extension, table_partition_cols
         )
 
-    def register_dataset(self, name: str, dataset: pa.dataset.Dataset) -> None:
-        """Register a :py:class:`pa.dataset.Dataset` as a table.
+    def register_dataset(self, name: str, dataset: object) -> None:
+        """Register any ``__arrow_c_stream__`` source as a table.
+
+        Any Python object implementing the Arrow ``__arrow_c_stream__`` protocol
+        can be registered, including objects from libraries such as nanoarrow,
+        Polars, DuckDB, or :py:mod:`pyarrow.dataset`.
 
         Args:
             name: Name of the table to register.
-            dataset: PyArrow dataset.
+            dataset: Object exposing ``__arrow_c_stream__``.
         """
         self.ctx.register_dataset(name, dataset)