From f06c4311103f8180a897c5e04c6e1f9c49ab4d4c Mon Sep 17 00:00:00 2001 From: Matthew Owen Date: Tue, 4 Mar 2025 12:00:34 -0800 Subject: [PATCH] adding in another kwargs Signed-off-by: Matthew Owen --- python/ray/air/util/tensor_extensions/arrow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/air/util/tensor_extensions/arrow.py b/python/ray/air/util/tensor_extensions/arrow.py index 52fade1a3917..81ffadbccc7e 100644 --- a/python/ray/air/util/tensor_extensions/arrow.py +++ b/python/ray/air/util/tensor_extensions/arrow.py @@ -516,7 +516,7 @@ def __arrow_ext_deserialize__(cls, storage_type, serialized): # TODO(Clark): Remove this version guard once we only support Arrow 9.0.0+. @PublicAPI(stability="beta") class ArrowTensorScalar(pa.ExtensionScalar): - def as_py(self) -> np.ndarray: + def as_py(self, **kwargs) -> np.ndarray: return self.type._extension_scalar_to_ndarray(self) def __array__(self) -> np.ndarray: