diff --git a/src/dataframe.rs b/src/dataframe.rs index 3b840735..2967a4ff 100644 --- a/src/dataframe.rs +++ b/src/dataframe.rs @@ -315,7 +315,6 @@ impl PyDataFrame { // Executes this DataFrame to get the total number of rows. fn count(&self, py: Python) -> PyResult { - let count = wait_for_future(py, self.df.as_ref().clone().count()); - Ok(count.unwrap()) + Ok(wait_for_future(py, self.df.as_ref().clone().count())?) } }