From 9e2540ef2109ea790284d002e9d2c6283eb808e2 Mon Sep 17 00:00:00 2001 From: Kyle Barron Date: Mon, 23 Jun 2025 15:34:29 -0700 Subject: [PATCH] Fix signature of `__arrow_c_stream__` --- python/datafusion/dataframe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/datafusion/dataframe.py b/python/datafusion/dataframe.py index 769271c7e..1fd63bdc6 100644 --- a/python/datafusion/dataframe.py +++ b/python/datafusion/dataframe.py @@ -1053,7 +1053,7 @@ def unnest_columns(self, *columns: str, preserve_nulls: bool = True) -> DataFram columns = list(columns) return DataFrame(self.df.unnest_columns(columns, preserve_nulls=preserve_nulls)) - def __arrow_c_stream__(self, requested_schema: pa.Schema) -> Any: + def __arrow_c_stream__(self, requested_schema: object | None = None) -> object: """Export an Arrow PyCapsule Stream. This will execute and collect the DataFrame. We will attempt to respect the