fix python asof join dispatch (#1720)

pola-rs · Nov 9, 2021 · a2bbd4d · a2bbd4d
1 parent 9c518f7
commit a2bbd4d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/py-polars/polars/eager/frame.py b/py-polars/polars/eager/frame.py
@@ -1969,6 +1969,12 @@ def join(
                 - "cross"
         suffix
             Suffix to append to columns with a duplicate name.
+        asof_by
+            join on these columns before doing asof join
+        asof_by_left
+            join on these columns before doing asof join
+        asof_by_right
+            join on these columns before doing asof join
         Returns
         -------
             Joined DataFrame

diff --git a/py-polars/polars/lazy/frame.py b/py-polars/polars/lazy/frame.py
@@ -464,6 +464,12 @@ def join(
             Allow the physical plan to optionally evaluate the computation of both DataFrames up to the join in parallel.
         force_parallel
             Force the physical plan to evaluate the computation of both DataFrames up to the join in parallel.
+        asof_by
+            join on these columns before doing asof join
+        asof_by_left
+            join on these columns before doing asof join
+        asof_by_right
+            join on these columns before doing asof join
 
         # Asof joins
         This is similar to a left-join except that we match on nearest key rather than equal keys.
@@ -535,7 +541,7 @@ def join(
         if isinstance(asof_by, str):
             left_asof_by_ = [asof_by]
             right_asof_by_ = [asof_by]
-        elif isinstance(on, list):
+        elif isinstance(asof_by, list):
             left_asof_by_ = asof_by
             right_asof_by_ = asof_by