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