g

apache · Aug 13, 2021 · 24cef6f · 24cef6f
1 parent 5d3eb09
commit 24cef6f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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]