lazy err on not found col (#3169)

pola-rs · Apr 17, 2022 · 0a14848 · 0a14848
1 parent e2f2828
commit 0a14848
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/polars/polars-lazy/src/logical_plan/optimizer/type_coercion.rs b/polars/polars-lazy/src/logical_plan/optimizer/type_coercion.rs
@@ -150,10 +150,10 @@ impl OptimizationRule for TypeCoercionRule {
 
                     let type_left = left
                         .get_type(input_schema, Context::Default, expr_arena)
-                        .expect("could not get dtype");
+                        .ok()?;
                     let type_right = right
                         .get_type(input_schema, Context::Default, expr_arena)
-                        .expect("could not get dtype");
+                        .ok()?;
 
                     #[allow(unused_mut, unused_assignments)]
                     let mut compare_cat_to_string = false;

diff --git a/polars/tests/it/lazy/projection_queries.rs b/polars/tests/it/lazy/projection_queries.rs
@@ -78,3 +78,17 @@ fn test_outer_join_with_column_2988() -> Result<()> {
 
     Ok(())
 }
+
+#[test]
+fn test_err_no_found() {
+    let df = df![
+        "a" => [1, 2, 3],
+        "b" => [None, Some("a"), Some("b")]
+    ]
+    .unwrap();
+
+    assert!(matches!(
+        df.lazy().filter(col("nope").gt(lit(2))).collect(),
+        Err(PolarsError::NotFound(_))
+    ));
+}