|
11 | 11 | from awkward._nplikes.numpy_like import NumpyMetadata
|
12 | 12 | from awkward._typing import Any, AxisMaybeNone, Literal
|
13 | 13 | from awkward.contents.content import ActionType, Content
|
| 14 | +from awkward.contents.remove_structure import remove_structure |
14 | 15 | from awkward.errors import AxisError
|
15 | 16 | from awkward.forms import form
|
16 | 17 | from awkward.record import Record
|
@@ -191,44 +192,6 @@ def pad_none(
|
191 | 192 | return layout._pad_none(length, axis, 1, clip)
|
192 | 193 |
|
193 | 194 |
|
194 |
| -def remove_structure( |
195 |
| - layout: Content | Record, |
196 |
| - backend: Backend | None = None, |
197 |
| - flatten_records: bool = True, |
198 |
| - function_name: str | None = None, |
199 |
| - drop_nones: bool = True, |
200 |
| - keepdims: bool = False, |
201 |
| - allow_records: bool = False, |
202 |
| - list_to_regular: bool = False, |
203 |
| -): |
204 |
| - if isinstance(layout, Record): |
205 |
| - return remove_structure( |
206 |
| - layout._array[layout._at : layout._at + 1], |
207 |
| - backend, |
208 |
| - flatten_records, |
209 |
| - function_name, |
210 |
| - drop_nones, |
211 |
| - keepdims, |
212 |
| - allow_records, |
213 |
| - ) |
214 |
| - |
215 |
| - else: |
216 |
| - if backend is None: |
217 |
| - backend = layout._backend |
218 |
| - arrays = layout._remove_structure( |
219 |
| - backend, |
220 |
| - { |
221 |
| - "flatten_records": flatten_records, |
222 |
| - "function_name": function_name, |
223 |
| - "drop_nones": drop_nones, |
224 |
| - "keepdims": keepdims, |
225 |
| - "allow_records": allow_records, |
226 |
| - "list_to_regular": list_to_regular, |
227 |
| - }, |
228 |
| - ) |
229 |
| - return tuple(arrays) |
230 |
| - |
231 |
| - |
232 | 195 | def flatten(layout: Content, axis: int = 1) -> Content:
|
233 | 196 | offsets, flattened = layout._offsets_and_flattened(axis, 1)
|
234 | 197 | return flattened
|
|
0 commit comments