fix doc test

apache · Feb 6, 2022 · 1b63a6b · 1b63a6b
1 parent 7cbbd5a
commit 1b63a6b
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 10 deletions.
diff --git a/datafusion-common/src/lib.rs b/datafusion-common/src/lib.rs
@@ -18,6 +18,8 @@
 mod column;
 mod dfschema;
 mod error;
+#[cfg(feature = "pyarrow")]
+mod pyarrow;
 mod scalar;
 
 pub use column::Column;

diff --git a/datafusion/src/pyarrow.rs → datafusion-common/src/pyarrow.rs b/datafusion/src/pyarrow.rs → datafusion-common/src/pyarrow.rs
@@ -15,12 +15,13 @@
 // specific language governing permissions and limitations
 // under the License.
 
-use pyo3::prelude::*;
-use pyo3::types::PyList;
+//! PyArrow
 
-use crate::arrow::array::ArrayData;
-use crate::arrow::pyarrow::PyArrowConvert;
-use crate::scalar::ScalarValue;
+use crate::ScalarValue;
+use arrow::array::ArrayData;
+use arrow::pyarrow::PyArrowConvert;
+use pyo3::types::PyList;
+use pyo3::{FromPyObject, IntoPy, PyAny, PyObject, PyResult, Python};
 
 impl PyArrowConvert for ScalarValue {
     fn from_pyarrow(value: &PyAny) -> PyResult<Self> {
@@ -68,7 +69,6 @@ mod tests {
     use pyo3::prepare_freethreaded_python;
     use pyo3::py_run;
     use pyo3::types::PyDict;
-    use pyo3::Python;
 
     fn init_python() {
         prepare_freethreaded_python();

diff --git a/datafusion-common/src/scalar.rs b/datafusion-common/src/scalar.rs
@@ -665,7 +665,7 @@ impl ScalarValue {
     ///
     /// Example
     /// ```
-    /// use datafusion::scalar::ScalarValue;
+    /// use datafusion_common::ScalarValue;
     /// use arrow::array::{ArrayRef, BooleanArray};
     ///
     /// let scalars = vec![

diff --git a/datafusion/src/lib.rs b/datafusion/src/lib.rs
@@ -224,9 +224,6 @@ pub use parquet;
 
 pub(crate) mod field_util;
 
-#[cfg(feature = "pyarrow")]
-mod pyarrow;
-
 pub mod from_slice;
 
 #[cfg(test)]