Skip to content

Commit 76c07f2

Browse files
committed
Correction after rebase
1 parent 2729fa0 commit 76c07f2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

crates/core/src/table.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -30,15 +30,15 @@ use datafusion::logical_expr::{
3030
use datafusion::physical_plan::ExecutionPlan;
3131
use datafusion::prelude::DataFrame;
3232
use datafusion_ffi::proto::logical_extension_codec::FFI_LogicalExtensionCodec;
33-
use datafusion_python_util::table_provider_from_pycapsule;
33+
use datafusion_python_util::{create_logical_extension_capsule, table_provider_from_pycapsule};
3434
use pyo3::IntoPyObjectExt;
3535
use pyo3::prelude::*;
3636

3737
use crate::context::PySessionContext;
3838
use crate::dataframe::PyDataFrame;
3939
use crate::dataset::Dataset;
40+
use crate::errors;
4041
use crate::expr::create_external_table::PyCreateExternalTable;
41-
use crate::{errors, utils};
4242

4343
/// This struct is used as a common method for all TableProviders,
4444
/// whether they refer to an FFI provider, an internally known
@@ -96,7 +96,7 @@ impl PyTable {
9696
Some(session) => session,
9797
None => PySessionContext::global_ctx()?.into_bound_py_any(obj.py())?,
9898
};
99-
utils::table_provider_from_pycapsule(obj.clone(), session)?
99+
table_provider_from_pycapsule(obj.clone(), session)?
100100
} {
101101
Ok(PyTable::from(provider))
102102
} else {
@@ -251,7 +251,7 @@ impl TableProviderFactory for RustWrappedPyTableProviderFactory {
251251
cmd: &CreateExternalTable,
252252
) -> datafusion::common::Result<Arc<dyn TableProvider>> {
253253
Python::attach(|py| {
254-
let codec = utils::create_logical_extension_capsule(py, self.codec.as_ref())
254+
let codec = create_logical_extension_capsule(py, self.codec.as_ref())
255255
.map_err(errors::to_datafusion_err)?;
256256

257257
self.create_inner(cmd.clone(), codec.into_any())

0 commit comments

Comments
 (0)