feat: make default buffer key nicer

dask-contrib · Oct 3, 2023 · 211b5ed · 211b5ed
1 parent 0f8b5d7
commit 211b5ed
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 11 deletions.
diff --git a/src/dask_awkward/lib/io/json.py b/src/dask_awkward/lib/io/json.py
@@ -82,7 +82,7 @@ def mock(self) -> AwkwardArray:
     def prepare_for_projection(
         self,
     ) -> tuple[AwkwardArray, TypeTracerReport, FormStructure]:
-        form = form_with_unique_keys(self.form, "<root>")
+        form = form_with_unique_keys(self.form, "@")
 
         # Build typetracer and associated report object
         (meta, report) = ak.typetracer.typetracer_with_report(

diff --git a/src/dask_awkward/lib/io/parquet.py b/src/dask_awkward/lib/io/parquet.py
@@ -80,7 +80,7 @@ def mock(self) -> AwkwardArray:
     def prepare_for_projection(
         self,
     ) -> tuple[AwkwardArray, TypeTracerReport, FormStructure]:
-        form = form_with_unique_keys(self.form, "<root>")
+        form = form_with_unique_keys(self.form, "@")
 
         # Build typetracer and associated report object
         (meta, report) = ak.typetracer.typetracer_with_report(

diff --git a/tests/test_inspect.py b/tests/test_inspect.py
@@ -16,9 +16,9 @@ def test_necessary_buffers(
         assert v == (
             frozenset(
                 {
-                    "<root>.points-offsets",
-                    "<root>.points.content.y-data",
-                    "<root>.points.content.x-data",
+                    "@.points-offsets",
+                    "@.points.content.y-data",
+                    "@.points.content.x-data",
                 }
             ),
             frozenset(),
@@ -29,7 +29,7 @@ def test_necessary_buffers(
     )
     for k, v in dak.report_necessary_buffers(w).items():
         assert v == (
-            frozenset({"<root>.points-offsets", "<root>.points.content.x-data"}),
+            frozenset({"@.points-offsets", "@.points.content.x-data"}),
             frozenset(),
         )
 
@@ -38,9 +38,9 @@ def test_necessary_buffers(
         assert v == (
             frozenset(
                 {
-                    "<root>.points-offsets",
-                    "<root>.points.content.x-data",
-                    "<root>.points.content.y-data",
+                    "@.points-offsets",
+                    "@.points.content.x-data",
+                    "@.points.content.y-data",
                 }
             ),
             frozenset(),
@@ -49,8 +49,8 @@ def test_necessary_buffers(
     z = dak.zeros_like(daa.points.x)
     for k, v in dak.report_necessary_buffers(z).items():
         assert v == (
-            frozenset({"<root>.points-offsets"}),
-            frozenset({"<root>.points.content.x-data"}),
+            frozenset({"@.points-offsets"}),
+            frozenset({"@.points.content.x-data"}),
         )