Update src/dataframe.rs

Co-authored-by: Tim Saucer <[email protected]>

Author: kosiew

src/dataframe.rs

@@ -753,7 +753,8 @@ impl PyDataFrame {
         let df = self.df.as_ref().clone();
         let fut: JoinHandle<datafusion::common::Result<SendableRecordBatchStream>> =
             rt.spawn(async move { df.execute_stream().await });
-        let stream = wait_for_future(py, async { fut.await.expect("Tokio task panicked") })??;
+        let stream = wait_for_future(py, async { fut.await })?
+            .map_err(|e| PyRuntimeError::new_err(format!("{e}")))??;
         Ok(PyRecordBatchStream::new(stream))
     }