fn read_final_stage()

in src/dataframe.rs [246:263]

• 18 lines of code
• 1 McCabe index (conditional complexity)

    fn read_final_stage(
        &mut self,
        py: Python,
        stage_id: usize,
        stage_addr: &str,
    ) -> PyResult<PyRecordBatchStream> {
        wait_for_future(
            py,
            collect_from_stage(
                stage_id,
                0,
                stage_addr,
                self.final_plan.take().unwrap().clone(),
            ),
        )
        .map(PyRecordBatchStream::new)
        .to_py_err()
    }