fix(polars): user newer drop API in asof join implementation

ibis-project · Feb 12, 2024 · c65d9f8 · c65d9f8
1 parent 43424ea
commit c65d9f8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ibis/backends/polars/compiler.py b/ibis/backends/polars/compiler.py
@@ -351,7 +351,10 @@ def asof_join(op, **kw):
 
     assert len(on) == 1
     joined = left.join_asof(right, on=on[0], by=by, strategy=direction)
-    joined = joined.drop(columns=on + by)
+    try:
+        joined = joined.drop(*(on + by))
+    except TypeError:
+        joined = joined.drop(columns=on + by)
     return joined