Skip to content

Commit b780838

Browse files
committed
fix: add stacklevel argument to deprecation warning in from_view method
1 parent 835cb20 commit b780838

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

src/table.rs

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ use datafusion::datasource::TableProvider;
2222
use datafusion_ffi::table_provider::{FFI_TableProvider, ForeignTableProvider};
2323
use pyo3::exceptions::PyDeprecationWarning;
2424
use pyo3::prelude::*;
25-
use pyo3::types::PyCapsule;
25+
use pyo3::types::{PyCapsule, PyDict};
2626

2727
use crate::catalog::PyTable;
2828
use crate::dataframe::PyDataFrame;
@@ -90,12 +90,15 @@ impl PyTableProvider {
9090
/// `TableProvider.from_dataframe` instead.
9191
#[staticmethod]
9292
pub fn from_view(py: Python<'_>, df: &PyDataFrame) -> PyDataFusionResult<Self> {
93-
py.import("warnings")?.call_method1(
93+
let kwargs = PyDict::new(py);
94+
kwargs.set_item("stacklevel", 2)?;
95+
py.import("warnings")?.call_method(
9496
"warn",
9597
(
9698
"PyTableProvider.from_view() is deprecated; use DataFrame.into_view() or TableProvider.from_dataframe() instead.",
9799
py.get_type::<PyDeprecationWarning>(),
98100
),
101+
Some(&kwargs),
99102
)?;
100103
Ok(Self::from_dataframe(df))
101104
}

0 commit comments

Comments
 (0)