fix(ir): compute InSubquery.shape property from needle input (#8364)

Fixes #8361.
ibis-project · Feb 15, 2024 · 13d675e · 13d675e
1 parent 2750fe5
commit 13d675e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/ibis/expr/operations/subqueries.py b/ibis/expr/operations/subqueries.py
@@ -53,7 +53,7 @@ class InSubquery(Subquery):
     needle: Value
 
     dtype = dt.boolean
-    shape = ds.columnar
+    shape = rlz.shape_like("needle")
 
     def __init__(self, rel, needle):
         if len(rel.schema) != 1:

diff --git a/ibis/tests/expr/test_value_exprs.py b/ibis/tests/expr/test_value_exprs.py
@@ -1713,3 +1713,13 @@ def test_deferred_doesnt_convert_callables():
         b=t.b.split(",").filter(lambda pp: ~pp.isin(("word1", "word2")))
     )
     assert expr.equals(expected)
+
+
+def test_in_subquery_shape():
+    t = ibis.table([("a", "int64"), ("b", "string")])
+
+    expr = t.a.cast("string").isin(t.b)
+    assert expr.op().shape.is_columnar()
+
+    expr = ibis.literal(2).isin(t.a)
+    assert expr.op().shape.is_scalar()