fix: properly catch not found explode cols (#17020)

pola-rs · Jun 17, 2024 · 69e3824 · 69e3824
1 parent 4a34528
commit 69e3824
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/crates/polars-plan/src/logical_plan/functions/dsl.rs b/crates/polars-plan/src/logical_plan/functions/dsl.rs
@@ -55,11 +55,13 @@ impl DslFunction {
                 let columns = columns
                     .iter()
                     .map(|e| {
-                        if let Expr::Column(name) = e {
-                            Ok(name.clone())
-                        } else {
+                        let Expr::Column(name) = e else {
                             polars_bail!(InvalidOperation: "expected column expression")
-                        }
+                        };
+
+                        polars_ensure!(input_schema.contains(name), ColumnNotFound: "{name}");
+
+                        Ok(name.clone())
                     })
                     .collect::<PolarsResult<Arc<[Arc<str>]>>>()?;
                 FunctionNode::Explode {

diff --git a/py-polars/tests/unit/operations/test_explode.py b/py-polars/tests/unit/operations/test_explode.py
@@ -430,3 +430,10 @@ def test_expr_str_explode_deprecated() -> None:
 
     expected = pl.Series("a", ["H", "e", "l", "l", "o", "W", "o", "r", "l", "d"])
     assert_series_equal(result, expected)
+
+
+def test_undefined_col_15852() -> None:
+    lf = pl.LazyFrame({"foo": [1]})
+
+    with pytest.raises(pl.exceptions.ColumnNotFoundError):
+        lf.explode("bar").join(lf, on="foo").collect()