Skip to content

Commit f85f7c6

Browse files
committed
python: Fix an issue with RawTableProvider
1 parent 4ea24b9 commit f85f7c6

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

python/src/lib.rs

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,8 @@ use std::{
33
sync::{Arc, OnceLock},
44
};
55

6-
use datafusion::catalog::TableProvider;
6+
use datafusion::{catalog::TableProvider, execution::TaskContextProvider, prelude::SessionContext};
77
use datafusion_ffi::table_provider::FFI_TableProvider;
8-
use datafusion_python::context::PySessionContext;
98
use pyo3::{prelude::*, types::PyCapsule};
109

1110
#[pyclass(module = "datafusion_table_providers._internal")]
@@ -34,8 +33,7 @@ impl RawTableProvider {
3433
None
3534
};
3635

37-
let ctx = Arc::new(PySessionContext::new(None, runtime));
38-
36+
let ctx = Arc::new(SessionContext::default()) as Arc<dyn TaskContextProvider>;
3937
let provider = FFI_TableProvider::new(
4038
Arc::clone(&self.table),
4139
self.supports_pushdown_filters,

0 commit comments

Comments
 (0)