feat(python): Release the GIL in collect_schema

pola-rs · Jul 21, 2024 · 0001a3f · 0001a3f
1 parent e8e68a1
commit 0001a3f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/py-polars/src/lazyframe/mod.rs b/py-polars/src/lazyframe/mod.rs
@@ -1193,7 +1193,9 @@ impl PyLazyFrame {
     }
 
     fn collect_schema(&mut self, py: Python) -> PyResult<PyObject> {
-        let schema = self.ldf.schema().map_err(PyPolarsErr::from)?;
+        let schema = py
+            .allow_threads(|| self.ldf.schema())
+            .map_err(PyPolarsErr::from)?;
 
         let schema_dict = PyDict::new_bound(py);
         schema.iter_fields().for_each(|fld| {