Fix: ensure posaxis is updated

scikit-hep · Jan 25, 2022 · dc458a4 · dc458a4
1 parent 8e06b8a
commit dc458a4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/awkward/_v2/operations/structure/ak_is_none.py b/src/awkward/_v2/operations/structure/ak_is_none.py
@@ -60,8 +60,10 @@ def getfunction_inner(layout, depth, **kwargs):
             return
 
     def getfunction_outer(layout, depth, depth_context, **kwargs):
-        posaxis = layout.axis_wrap_if_negative(depth_context["posaxis"])
-        if posaxis != depth - 1:
+        depth_context["posaxis"] = layout.axis_wrap_if_negative(
+            depth_context["posaxis"]
+        )
+        if depth_context["posaxis"] != depth - 1:
             return
 
         return layout.recursively_apply(getfunction_inner)