fix(polars): user newer drop API to avoid deprecation warning

ibis-project · Feb 12, 2024 · 43424ea · 43424ea
1 parent c836339
commit 43424ea
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ibis/backends/polars/compiler.py b/ibis/backends/polars/compiler.py
@@ -313,7 +313,11 @@ def join(op, **kw):
         left, right = right, left
 
     joined = left.join(right, on=on, how=how)
-    joined = joined.drop(columns=on)
+
+    try:
+        joined = joined.drop(*on)
+    except TypeError:
+        joined = joined.drop(columns=on)
 
     return joined