fix: support typetracer in to_packed for IndexedOptionArray

scikit-hep · Dec 21, 2023 · b408feb · b408feb
1 parent a9c2e3b
commit b408feb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/awkward/contents/indexedoptionarray.py b/src/awkward/contents/indexedoptionarray.py
@@ -1721,8 +1721,10 @@ def to_packed(self) -> Self:
         original_index = self._index.raw(self._backend.nplike)
         is_none = original_index < 0
         num_none = self._backend.index_nplike.count_nonzero(is_none)
-        if self.parameter("__array__") == "categorical" or self._content.length <= (
-            len(original_index) - num_none
+        if (
+            self.parameter("__array__") == "categorical"
+            or not self._backend.index_nplike.known_data
+            or self._content.length <= (len(original_index) - num_none)
         ):
             return ak.contents.IndexedOptionArray(
                 self._index, self._content.to_packed(), parameters=self._parameters