diff --git a/python/ray/air/util/object_extensions/arrow.py b/python/ray/air/util/object_extensions/arrow.py index a56a04869855..fc38fde32e50 100644 --- a/python/ray/air/util/object_extensions/arrow.py +++ b/python/ray/air/util/object_extensions/arrow.py @@ -77,7 +77,7 @@ def __reduce__(self): class ArrowPythonObjectScalar(pa.ExtensionScalar): """Scalar class for ArrowPythonObjectType""" - def as_py(self) -> typing.Any: + def as_py(self, **kwargs) -> typing.Any: if not isinstance(self.value, pa.LargeBinaryScalar): raise RuntimeError( f"{type(self.value)} is not the expected LargeBinaryScalar"