fix(trino): only support how='first' with arbitrary reduction

ibis-project · Mar 26, 2023 · 315b5e7 · 315b5e7
1 parent 1a16401
commit 315b5e7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ibis/backends/trino/registry.py b/ibis/backends/trino/registry.py
@@ -48,8 +48,10 @@ def _literal(t, op):
 
 
 def _arbitrary(t, op):
-    if op.how == "heavy":
-        raise com.UnsupportedOperationError('Trino does not support how="heavy"')
+    if op.how != "first":
+        raise com.UnsupportedOperationError(
+            'Trino only supports how="first" for `arbitrary` reduction'
+        )
     return reduction(sa.func.arbitrary)(t, op)