python polars 0.13.62 (#4294)

* join expression argument fix * python polars 0.13.62
pola-rs · Aug 6, 2022 · 09bb034 · 09bb034
1 parent 093d55c
commit 09bb034
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 6 deletions.
diff --git a/polars/polars-lazy/src/logical_plan/alp.rs b/polars/polars-lazy/src/logical_plan/alp.rs
@@ -799,10 +799,9 @@ impl<'a> ALogicalPlanBuilder<'a> {
 
         let right_names: PlHashSet<_> = right_on
             .iter()
-            .map(|e| match self.expr_arena.get(*e) {
-                AExpr::Alias(_, name) => name.clone(),
-                AExpr::Column(name) => name.clone(),
-                _ => panic!("could not determine join column names"),
+            .map(|e| {
+                aexpr_to_root_column_name(*e, self.expr_arena)
+                    .expect("could not determine join column names")
             })
             .collect();
 

diff --git a/py-polars/Cargo.lock b/py-polars/Cargo.lock
diff --git a/py-polars/Cargo.toml b/py-polars/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "py-polars"
-version = "0.13.61"
+version = "0.13.62"
 authors = ["ritchie46 <ritchie46@gmail.com>"]
 documentation = "https://pola-rs.github.io/polars/py-polars/html/reference/index.html"
 edition = "2021"

diff --git a/py-polars/tests/test_joins.py b/py-polars/tests/test_joins.py
@@ -301,3 +301,6 @@ def test_join_on_cast() -> None:
         "row_nr": [1, 2, 3, 5],
         "a": [-2, 3, 3, 10],
     }
+    assert df_a.lazy().join(
+        df_b.lazy(), on=pl.col("a").cast(pl.Int64)
+    ).collect().to_dict(False) == {"row_nr": [1, 2, 3, 5], "a": [-2, 3, 3, 10]}