Export at top level

kylebarron · Jun 25, 2024 · 9c40057 · 9c40057
1 parent 5e2a363
commit 9c40057
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 19 deletions.
diff --git a/arro3-compute/src/concat.rs b/arro3-compute/src/concat.rs
@@ -1,7 +1,6 @@
 use pyo3::prelude::*;
-use pyo3_arrow::array::PyArray;
-use pyo3_arrow::chunked::PyChunkedArray;
 use pyo3_arrow::error::PyArrowResult;
+use pyo3_arrow::{PyArray, PyChunkedArray};
 
 #[pyfunction]
 pub fn concat(input: PyChunkedArray) -> PyArrowResult<PyObject> {

diff --git a/arro3-compute/src/take.rs b/arro3-compute/src/take.rs
@@ -1,6 +1,6 @@
 use pyo3::prelude::*;
-use pyo3_arrow::array::PyArray;
 use pyo3_arrow::error::PyArrowResult;
+use pyo3_arrow::PyArray;
 
 #[pyfunction]
 pub fn take(values: PyArray, indices: PyArray) -> PyArrowResult<PyObject> {

diff --git a/arro3-core/src/lib.rs b/arro3-core/src/lib.rs
@@ -12,13 +12,13 @@ fn ___version() -> &'static str {
 fn _rust(_py: Python, m: &Bound<PyModule>) -> PyResult<()> {
     m.add_wrapped(wrap_pyfunction!(___version))?;
 
-    m.add_class::<pyo3_arrow::array::PyArray>()?;
-    m.add_class::<pyo3_arrow::chunked::PyChunkedArray>()?;
-    m.add_class::<pyo3_arrow::field::PyField>()?;
-    m.add_class::<pyo3_arrow::record_batch::PyRecordBatch>()?;
-    m.add_class::<pyo3_arrow::record_batch_reader::PyRecordBatchReader>()?;
-    m.add_class::<pyo3_arrow::schema::PySchema>()?;
-    m.add_class::<pyo3_arrow::table::PyTable>()?;
+    m.add_class::<pyo3_arrow::PyArray>()?;
+    m.add_class::<pyo3_arrow::PyChunkedArray>()?;
+    m.add_class::<pyo3_arrow::PyField>()?;
+    m.add_class::<pyo3_arrow::PyRecordBatch>()?;
+    m.add_class::<pyo3_arrow::PyRecordBatchReader>()?;
+    m.add_class::<pyo3_arrow::PySchema>()?;
+    m.add_class::<pyo3_arrow::PyTable>()?;
 
     Ok(())
 }
diff --git a/pyo3-arrow/src/lib.rs b/pyo3-arrow/src/lib.rs
@@ -1,12 +1,20 @@
 #![doc = include_str!("../README.md")]
 
-pub mod array;
-pub mod chunked;
+mod array;
+mod chunked;
 pub mod error;
-pub mod ffi;
-pub mod field;
-pub mod interop;
-pub mod record_batch;
-pub mod record_batch_reader;
-pub mod schema;
-pub mod table;
+mod ffi;
+mod field;
+mod interop;
+mod record_batch;
+mod record_batch_reader;
+mod schema;
+mod table;
+
+pub use array::PyArray;
+pub use chunked::PyChunkedArray;
+pub use field::PyField;
+pub use record_batch::PyRecordBatch;
+pub use record_batch_reader::PyRecordBatchReader;
+pub use schema::PySchema;
+pub use table::PyTable;