From 24cef6f216057915225b7bff553be59b8ae3599a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kriszti=C3=A1n=20Sz=C5=B1cs?= <szucs.krisztian@gmail.com>
Date: Fri, 13 Aug 2021 13:32:16 +0200
Subject: [PATCH] g

---
 arrow-pyarrow-integration-testing/src/lib.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/arrow-pyarrow-integration-testing/src/lib.rs b/arrow-pyarrow-integration-testing/src/lib.rs
index dfa767821548..659cdeccb577 100644
--- a/arrow-pyarrow-integration-testing/src/lib.rs
+++ b/arrow-pyarrow-integration-testing/src/lib.rs
@@ -118,14 +118,14 @@ fn substring(array: ArrayData, start: i64) -> PyResult<ArrayData> {
 
 /// Returns the concatenate
 #[pyfunction]
-fn concatenate(array: ArrayData) -> PyResult<ArrayData> {
+fn concatenate(array: ArrayData, py: Python) -> PyResult<PyObject> {
     let array = ArrayRef::from(array);
 
     // concat
     let array = kernels::concat::concat(&[array.as_ref(), array.as_ref()])
         .map_err(PyO3ArrowError::from)?;
 
-    Ok(array.data().to_owned())
+    array.to_pyarrow(py)
 }
 
 #[pyfunction]