feat: allow selection subclasses to define new node args

ibis-project · Jun 22, 2022 · 2a7dc41 · 2a7dc41
1 parent 0062429
commit 2a7dc41
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ibis/expr/operations/relations.py b/ibis/expr/operations/relations.py
@@ -379,7 +379,7 @@ class Selection(TableNode, sch.HasSchema):
         default=(),
     )
 
-    def __init__(self, table, selections, predicates, sort_keys):
+    def __init__(self, table, selections, predicates, sort_keys, **kwargs):
         from ibis.expr.analysis import shares_all_roots, shares_some_roots
 
         if not shares_all_roots(selections + sort_keys, table):
@@ -399,6 +399,7 @@ def __init__(self, table, selections, predicates, sort_keys):
             selections=selections,
             predicates=predicates,
             sort_keys=sort_keys,
+            **kwargs,
         )
 
         # Validate no overlapping columns in schema