Update pyarrow method call with kwargs (#3560)

apache · Jan 19, 2023 · d980235 · d980235
1 parent de62808
commit d980235
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/arrow/src/pyarrow.rs b/arrow/src/pyarrow.rs
@@ -24,7 +24,7 @@ use std::sync::Arc;
 use pyo3::ffi::Py_uintptr_t;
 use pyo3::import_exception;
 use pyo3::prelude::*;
-use pyo3::types::{PyList, PyTuple};
+use pyo3::types::{PyDict, PyList, PyTuple};
 
 use crate::array::{make_array, Array, ArrayData};
 use crate::datatypes::{DataType, Field, Schema};
@@ -196,8 +196,10 @@ impl PyArrowConvert for RecordBatch {
 
         let module = py.import("pyarrow")?;
         let class = module.getattr("RecordBatch")?;
-        let record = class
-            .call_method1("from_arrays", (py_arrays, None::<PyObject>, py_schema))?;
+        let args = (py_arrays,);
+        let kwargs = PyDict::new(py);
+        kwargs.set_item("schema", py_schema)?;
+        let record = class.call_method("from_arrays", args, Some(kwargs))?;
 
         Ok(PyObject::from(record))
     }