feat: add Arrow interoperability support to RecordBatch classes

Author: kosiew

python/datafusion/record_batch.py

@@ -46,6 +46,14 @@ def to_pyarrow(self) -> pa.RecordBatch:
         """Convert to :py:class:`pa.RecordBatch`."""
         return self.record_batch.to_pyarrow()
 
+    def __arrow__(self, *args: object, **kwargs: object) -> pa.RecordBatch:
+        """Return a :py:class:`pa.RecordBatch` for Arrow interoperability.
+
+        This enables ``datafusion.record_batch.RecordBatch`` instances to be
+        automatically recognized by PyArrow when passed to its APIs.
+        """
+        return self.to_pyarrow()
+
     def __arrow_c_array__(
         self, requested_schema: object | None = None
     ) -> tuple[object, object]:

src/record_batch.rs

@@ -79,6 +79,10 @@ impl PyRecordBatch {
         self.batch.to_pyarrow(py)
     }
 
+    fn __arrow__(&self, py: Python) -> PyResult<PyObject> {
+        self.to_pyarrow(py)
+    }
+
     #[pyo3(signature = (requested_schema=None))]
     fn __arrow_c_array__<'py>(
         &self,