handle error from wait_for_future in count method of PyDataFrame

kosiew · kosiew · commit 84f31bbaa88c · 2025-06-04T17:16:13.000+08:00
diff --git a/src/dataframe.rs b/src/dataframe.rs
@@ -864,7 +864,7 @@ impl PyDataFrame {
 
     // Executes this DataFrame to get the total number of rows.
     fn count(&self, py: Python) -> PyDataFusionResult<usize> {
-        Ok(wait_for_future(py, self.df.as_ref().clone().count())??)
+        Ok(wait_for_future(py, self.df.as_ref().clone().count()).map_err(py_datafusion_err)??)
     }
 
     /// Fill null values with a specified value for specific columns

Original file line number	Diff line number	Diff line change
`@@ -864,7 +864,7 @@ impl PyDataFrame {`
`864`	`864`
`865`	`865`	`// Executes this DataFrame to get the total number of rows.`
`866`	`866`	`fn count(&self, py: Python) -> PyDataFusionResult<usize> {`
`867`		`- Ok(wait_for_future(py, self.df.as_ref().clone().count())??)`
	`867`	`+ Ok(wait_for_future(py, self.df.as_ref().clone().count()).map_err(py_datafusion_err)??)`
`868`	`868`	`}`
`869`	`869`
`870`	`870`	`/// Fill null values with a specified value for specific columns`