Add dak.to_list

src/dask_awkward/__init__.py

@@ -90,6 +90,7 @@
     singletons,
     sort,
     strings_astype,
+    to_list,
     to_packed,
     to_regular,
     unflatten,

src/dask_awkward/lib/__init__.py

@@ -76,6 +76,7 @@
     singletons,
     sort,
     strings_astype,
+    to_list,
     to_packed,
     to_regular,
     unflatten,

src/dask_awkward/lib/structure.py

@@ -62,6 +62,7 @@
     "singletons",
     "sort",
     "strings_astype",
+    "to_list",
     "to_packed",
     "to_regular",
     "unflatten",
@@ -691,6 +692,15 @@ def ones_like(
     )
 
 
+@borrow_docstring(ak.to_list)
+def to_list(array: Array) -> list:
+    """Return list/dict version of the data
+
+    Unlike most functions, this one requires a compute() of the data.
+    """
+    return array.compute().to_list()
+
+
 @borrow_docstring(ak.to_packed)
 def to_packed(
     array: Array,

tests/test_behavior.py

@@ -66,9 +66,6 @@ def test_distance_behavior(
     caa1 = ak.Array(caa_p1.points, with_name="Point", behavior=behaviors)
     caa2 = ak.Array(caa_p2.points)
 
-    print(f"{ak.to_list(daa1.distance(daa2)) = }")
-    print(f"{ak.to_list(caa1.distance(caa2)) = }")
-
     assert_eq(daa1.distance(daa2), caa1.distance(caa2))
     assert_eq(np.abs(daa1), np.abs(caa1))
 

tests/test_structure.py

@@ -458,6 +458,10 @@ def test_to_packed(daa, caa):
     )
 
 
+def test_to_list(daa):
+    assert dak.to_list(daa) == daa.compute().to_list()
+
+
 def test_ravel(daa, caa):
     assert_eq(
         dak.ravel(daa.points.x),