Feature: add pass_apply argument to util

scikit-hep · Jun 11, 2021 · 43f6fcf · 43f6fcf
1 parent 94de4e5
commit 43f6fcf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/awkward/_util.py b/src/awkward/_util.py
@@ -1100,6 +1100,7 @@ def recursively_apply(
     getfunction,
     pass_depth=True,
     pass_user=False,
+    pass_apply=False,
     user=None,
     keep_parameters=True,
     numpy_to_regular=False,
@@ -1113,6 +1114,8 @@ def apply(layout, depth, user):
             args = args + (depth,)
         if pass_user:
             args = args + (user,)
+        if pass_apply:
+            args = args + (apply,)
 
         custom = getfunction(layout, *args)
         if callable(custom):