High-level to/from_arrow functions for v2.

scikit-hep · Nov 3, 2021 · 151508c · 151508c
1 parent 81b55a2
commit 151508c
Show file tree

Hide file tree

Showing 7 changed files with 263 additions and 1,018 deletions.
diff --git a/src/awkward/_v2/contents/content.py b/src/awkward/_v2/contents/content.py
@@ -1233,7 +1233,7 @@ def to_arrow(
         string_to32=False,
         bytestring_to32=False,
         emptyarray_to=None,
-        indexedarray_as_dictionary=False,
+        categorical_as_dictionary=False,
         extensionarray=True,
         count_nulls=True,
     ):
@@ -1250,7 +1250,7 @@ def to_arrow(
                 "string_to32": string_to32,
                 "bytestring_to32": bytestring_to32,
                 "emptyarray_to": emptyarray_to,
-                "indexedarray_as_dictionary": indexedarray_as_dictionary,
+                "categorical_as_dictionary": categorical_as_dictionary,
                 "extensionarray": extensionarray,
                 "count_nulls": count_nulls,
             },

diff --git a/src/awkward/_v2/contents/indexedarray.py b/src/awkward/_v2/contents/indexedarray.py
@@ -729,7 +729,7 @@ def _validityerror(self, path):
 
     def _to_arrow(self, pyarrow, mask_node, validbytes, length, options):
         if (
-            not options["indexedarray_as_dictionary"]
+            not options["categorical_as_dictionary"]
             and self.parameter("__array__") == "categorical"
         ):
             next_parameters = dict(self._parameters)