error conversion

Author: ccciudatu

src/dataframe.rs

@@ -663,7 +663,7 @@ impl PyDataFrame {
 
     fn distributed_plan(&self, num_shards: usize, py: Python<'_>) -> PyResult<DistributedPlan> {
         let distributed_plan = wait_for_future(py, split_physical_plan(&self.df, num_shards))
-            .map_err(PyErr::from)?;
+            .map_err(py_datafusion_err)?;
         Ok(distributed_plan)
     }
 
@@ -802,11 +802,11 @@ pub fn shard_stream(serialized_shard_plan: &[u8]) -> PyResult<PyRecordBatchStrea
     let codec = DeltaPhysicalCodec {};
     let node = PhysicalPlanNode::decode(serialized_shard_plan)
         .map_err(|e| datafusion::error::DataFusionError::External(Box::new(e)))
-        .map_err(PyErr::from)?;
+        .map_err(py_datafusion_err)?;
     let plan = node.try_into_physical_plan(&registry, &runtime, &codec)?;
     println!("Shard plan: {}", displayable(plan.as_ref()).one_line());
     let ctx = TaskContext::default();
-    execute_stream(plan, Arc::new(ctx)).map(PyRecordBatchStream::new).map_err(PyErr::from)
+    execute_stream(plan, Arc::new(ctx)).map(PyRecordBatchStream::new).map_err(py_datafusion_err)
 }
 
 /// Print DataFrame