chore: add type annotation for ArrayValue.filter method input

ibis-project · Jul 17, 2023 · 603cb6a · 603cb6a · ibis-squawk-bot · Jul 17, 2023
1 parent 3355dd8
commit 603cb6a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ibis/expr/types/arrays.py b/ibis/expr/types/arrays.py
@@ -430,7 +430,9 @@ def map(self, func: Callable[[ir.Value], ir.Value]) -> ir.ArrayValue:
         """
         return ops.ArrayMap(self, func=func).to_expr()
 
-    def filter(self, predicate: Callable[[ir.Value], ir.BooleanValue]) -> ir.ArrayValue:
+    def filter(
+        self, predicate: Callable[[ir.Value], bool | ir.BooleanValue]
+    ) -> ir.ArrayValue:
         """Filter array elements using `predicate`.
 
         Parameters