fix: support indexed types in fill_none

scikit-hep · Jan 12, 2023 · ff67e0f · ff67e0f
1 parent 29712f5
commit ff67e0f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/awkward/operations/ak_fill_none.py b/src/awkward/operations/ak_fill_none.py
@@ -117,10 +117,10 @@ def action(layout, continuation, **kwargs):
         def action(layout, depth, **kwargs):
             posaxis = ak._util.maybe_posaxis(layout, axis, depth)
             if posaxis is not None and posaxis + 1 == depth:
-                if layout.is_union or layout.is_record:
-                    return None
-                elif layout.is_option:
+                if layout.is_option:
                     return ak._do.fill_none(layout, valuelayout)
+                elif layout.is_union or layout.is_record or layout.is_indexed:
+                    return None
                 else:
                     return layout