refactor(rust): Change fn schema to fn collect_schema

crates/polars-lazy/src/frame/cached_arenas.rs

@@ -36,7 +36,7 @@ impl LazyFrame {
     ///
     /// Returns an `Err` if the logical plan has already encountered an error (i.e., if
     /// `self.collect()` would fail), `Ok` otherwise.
-    pub fn schema(&mut self) -> PolarsResult<SchemaRef> {
+    pub fn collect_schema(&mut self) -> PolarsResult<SchemaRef> {
         let mut cached_arenas = self.cached_arena.lock().unwrap();
 
         match &mut *cached_arenas {

crates/polars-lazy/src/frame/mod.rs

@@ -1045,7 +1045,7 @@ impl LazyFrame {
             options.index_column = name.as_ref().into();
         } else {
             let output_field = index_column
-                .to_field(&self.schema().unwrap(), Context::Default)
+                .to_field(&self.collect_schema().unwrap(), Context::Default)
                 .unwrap();
             return self.with_column(index_column).rolling(
                 Expr::Column(Arc::from(output_field.name().as_str())),
@@ -1090,7 +1090,7 @@ impl LazyFrame {
             options.index_column = name.as_ref().into();
         } else {
             let output_field = index_column
-                .to_field(&self.schema().unwrap(), Context::Default)
+                .to_field(&self.collect_schema().unwrap(), Context::Default)
                 .unwrap();
             return self.with_column(index_column).group_by_dynamic(
                 Expr::Column(Arc::from(output_field.name().as_str())),

crates/polars-lazy/src/tests/optimization_checks.rs

@@ -495,7 +495,7 @@ fn test_with_column_prune() -> PolarsResult<()> {
         matches!(lp, SimpleProjection { .. } | DataFrameScan { .. })
     }));
     assert_eq!(
-        q.schema().unwrap().as_ref(),
+        q.collect_schema().unwrap().as_ref(),
         &Schema::from_iter([Field::new("c1", DataType::Int32)])
     );
     Ok(())

crates/polars-lazy/src/tests/queries.rs

@@ -1440,7 +1440,7 @@ fn test_when_then_schema() -> PolarsResult<()> {
         .select([when(col("A").gt(lit(1)))
             .then(Null {}.lit())
             .otherwise(col("A"))])
-        .schema();
+        .collect_schema();
     assert_ne!(schema?.get_at_index(0).unwrap().1, &DataType::Null);
 
     Ok(())

crates/polars-python/src/lazyframe/general.rs

@@ -1170,7 +1170,7 @@ impl PyLazyFrame {
 
     fn collect_schema(&mut self, py: Python) -> PyResult<PyObject> {
         let schema = py
-            .allow_threads(|| self.ldf.schema())
+            .allow_threads(|| self.ldf.collect_schema())
             .map_err(PyPolarsErr::from)?;
 
         let schema_dict = PyDict::new_bound(py);

crates/polars-python/src/lazygroupby.rs

@@ -43,7 +43,7 @@ impl PyLazyGroupBy {
         let schema = match schema {
             Some(schema) => Arc::new(schema.0),
             None => LazyFrame::from(lgb.logical_plan.clone())
-                .schema()
+                .collect_schema()
                 .map_err(PyPolarsErr::from)?,
         };