fix: index with int in unionarray.py (#3419)

ianna · web-flow · commit c20bd115a09a · 2025-03-19T17:42:19.000+01:00
* fix index with int in unionarray.py

* Update src/awkward/contents/unionarray.py
diff --git a/src/awkward/contents/unionarray.py b/src/awkward/contents/unionarray.py
@@ -1604,7 +1604,7 @@ def _remove_structure(
         # backends with concrete data
         for i in range(self._tags.length):
             content = (
-                self._contents[self._tags.data[i]]
+                self._contents[self._tags[i]]
                 ._carry(ak.index.Index(self._index.data[i]), False)
                 ._remove_structure(backend, options)
             )

Original file line number	Diff line number	Diff line change
`@@ -1604,7 +1604,7 @@ def _remove_structure(`
`1604`	`1604`	`# backends with concrete data`
`1605`	`1605`	`for i in range(self._tags.length):`
`1606`	`1606`	`content = (`
`1607`		`- self._contents[self._tags.data[i]]`
	`1607`	`+ self._contents[self._tags[i]]`
`1608`	`1608`	`._carry(ak.index.Index(self._index.data[i]), False)`
`1609`	`1609`	`._remove_structure(backend, options)`
`1610`	`1610`	`)`