move py_err_to_datafusion_err to errors.rs

Author: kosiew

src/context.rs

@@ -34,7 +34,9 @@ use pyo3::prelude::*;
 use crate::catalog::{PyCatalog, PyTable};
 use crate::dataframe::PyDataFrame;
 use crate::dataset::Dataset;
-use crate::errors::{py_datafusion_err, PyDataFusionError, PyDataFusionResult};
+use crate::errors::{
+    py_datafusion_err, py_err_to_datafusion_err, PyDataFusionError, PyDataFusionResult,
+};
 use crate::expr::sort_expr::PySortExpr;
 use crate::physical_plan::PyExecutionPlan;
 use crate::record_batch::PyRecordBatchStream;
@@ -1438,11 +1440,6 @@ pub fn parse_file_compression_type(
         })
 }
 
-/// Convert a PyErr to a DataFusionError
-pub fn py_err_to_datafusion_err(err: PyErr) -> datafusion::error::DataFusionError {
-    datafusion::error::DataFusionError::Execution(format!("Python error: {}", err))
-}
-
 impl From<PySessionContext> for SessionContext {
     fn from(ctx: PySessionContext) -> SessionContext {
         ctx.ctx

src/errors.rs

@@ -95,3 +95,8 @@ pub fn py_unsupported_variant_err(e: impl Debug) -> PyErr {
 pub fn to_datafusion_err(e: impl Debug) -> InnerDataFusionError {
     InnerDataFusionError::Execution(format!("{e:?}"))
 }
+
+/// Convert a PyErr to a DataFusionError
+pub fn py_err_to_datafusion_err(err: PyErr) -> InnerDataFusionError {
+    InnerDataFusionError::Execution(format!("Python error: {}", err))
+}