@@ -3,38 +3,33 @@ using PythonCall
33using JSON
44import AwkwardArray
55
6- function _as_numpy (array:: AbstractVector{UInt8} )
7- np = pyimport (" numpy" )
8- np. asarray (array, dtype = np. uint8)
9- end
10-
11- function AwkwardArray. julia_array_to_python (array)
12- form, len, containers = AwkwardArray. to_buffers (array)
6+ function AwkwardArray. convert (layout:: AwkwardArray.Content ):: Py
7+ form, len, containers = AwkwardArray. to_buffers (layout)
138
149 py_buffers = Dict {String,Any} ()
1510
1611 for (key, buffer) in containers
17- py_buffers[key] = _as_numpy ( buffer)
12+ py_buffers[key] = pyimport ( " numpy " ) . asarray ( buffer, dtype = pyimport ( " numpy " ) . uint8 )
1813 end
1914
20- return pyimport (" awkward" ). from_buffers (form, len, py_buffers)
15+ pyimport (" awkward" ). from_buffers (form, len, py_buffers)
2116end
2217
23- function _as_julia (py_buffer)
24- uint8_buffer = reinterpret (UInt8, py_buffer)
25- return uint8_buffer
26- end
27-
28- function AwkwardArray. python_array_to_julia (py_array)
29- form, len, _containers = pyimport (" awkward" ). to_buffers (py_array)
18+ function AwkwardArray. convert (array:: Py ):: AwkwardArray.Content
19+ form, len, _containers = pyimport (" awkward" ). to_buffers (array)
3020 containers = pyconvert (Dict, _containers)
3121
3222 julia_buffers = Dict {String,AbstractVector{UInt8}} ()
23+
3324 for (key, buffer) in containers
34- julia_buffers[key] = _as_julia ( buffer)
25+ julia_buffers[key] = reinterpret (UInt8, buffer)
3526 end
3627
37- return AwkwardArray. from_buffers (pyconvert (String, form. to_json ()), pyconvert (Int, len), julia_buffers)
28+ AwkwardArray. from_buffers (
29+ pyconvert (String, form. to_json ()),
30+ pyconvert (Int, len),
31+ julia_buffers,
32+ )
3833end
3934
4035end # module
0 commit comments