add Primitive and ListOffsetArray adapts (#119)

Author: Moelf

Project.toml

@@ -9,12 +9,15 @@ JSON = "682c06a0-de6a-54ab-a142-c8b1cf79cde6"
 Tables = "bd369af6-aec1-5ad0-b16a-f7cc5008161c"
 
 [weakdeps]
+Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 PythonCall = "6099a3de-0909-46bc-b1f4-468b9a2dfc0d"
 
 [extensions]
 AwkwardPythonCallExt = "PythonCall"
+AwkwardAdaptExt = "Adapt"
 
 [compat]
+Adapt = "4.3.0"
 Conda = "1.10.0"
 JSON = "0.21.4"
 PythonCall = "0.9"

ext/AwkwardAdaptExt.jl

@@ -0,0 +1,16 @@
+module AwkwardAdaptExt
+
+import Adapt
+import AwkwardArray as ak
+
+function Adapt.adapt_structure(to, x::ak.PrimitiveArray{I, T, B}) where {I, T, B}
+    ak.PrimitiveArray(Adapt.adapt(to, x.data); x.parameters, behavior = B)
+end
+
+function Adapt.adapt_structure(to, x::ak.ListOffsetArray{I, T, B}) where {I, T, B}
+    new_offsets = Adapt.adapt(to, x.offsets)
+    new_content = Adapt.adapt(to, x.content)
+    ak.ListOffsetArray(new_offsets, new_content; x.parameters, behavior = B)
+end
+
+end