refactor: add type annotations to set operation functions

ibis-project · Sep 28, 2023 · 13f593b · 13f593b
1 parent c0085be
commit 13f593b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ibis/expr/api.py b/ibis/expr/api.py
@@ -1531,7 +1531,7 @@ def trailing_range_window(preceding, order_by, group_by=None):
     )
 
 
-def union(table: ir.Table, *rest: ir.Table, distinct: bool = False):
+def union(table: ir.Table, *rest: ir.Table, distinct: bool = False) -> ir.Table:
     """Compute the set union of multiple table expressions.
 
     The input tables must have identical schemas.
@@ -1599,7 +1599,7 @@ def union(table: ir.Table, *rest: ir.Table, distinct: bool = False):
     return table.union(*rest, distinct=distinct) if rest else table
 
 
-def intersect(table: ir.Table, *rest: ir.Table, distinct: bool = True):
+def intersect(table: ir.Table, *rest: ir.Table, distinct: bool = True) -> ir.Table:
     """Compute the set intersection of multiple table expressions.
 
     The input tables must have identical schemas.
@@ -1654,7 +1654,7 @@ def intersect(table: ir.Table, *rest: ir.Table, distinct: bool = True):
     return table.intersect(*rest, distinct=distinct) if rest else table
 
 
-def difference(table: ir.Table, *rest: ir.Table, distinct: bool = True):
+def difference(table: ir.Table, *rest: ir.Table, distinct: bool = True) -> ir.Table:
     """Compute the set difference of multiple table expressions.
 
     The input tables must have identical schemas.