Allow old pyarrow to forget that an unnullable string has unnullable …

…outputs.
scikit-hep · Aug 15, 2023 · dfa5d06 · dfa5d06
1 parent 7ebf8ff
commit dfa5d06
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/awkward/operations/str/__init__.py b/src/awkward/operations/str/__init__.py
@@ -119,6 +119,9 @@ def action(layout, **absorb):
                     ),
                     highlevel=False,
                 )
+
+                if out.is_option:
+                    out = out.content
                 if out.is_list and out.parameter("__array__") == "string":
                     out = out.copy(
                         content=out.content.copy(parameters={"__array__": "byte"}),
@@ -140,6 +143,9 @@ def action(layout, **absorb):
 def _erase_list_option(layout):
     from awkward.contents.unmaskedarray import UnmaskedArray
 
+    if layout.is_option:
+        layout = layout.content
+
     assert layout.is_list
     if layout.content.is_option:
         assert isinstance(layout.content, UnmaskedArray)