Revert "fix: drop lengths recursively (#2775)"

This reverts commit 8f2285a.
scikit-hep · Nov 1, 2023 · 21e5ae7 · 21e5ae7
1 parent aed7f20
commit 21e5ae7
Show file tree

Hide file tree

Showing 8 changed files with 7 additions and 22 deletions.
diff --git a/src/awkward/contents/bitmaskedarray.py b/src/awkward/contents/bitmaskedarray.py
@@ -311,7 +311,7 @@ def _to_typetracer(self, forget_length: bool) -> Self:
         tt = TypeTracer.instance()
         return BitMaskedArray(
             self._mask.to_nplike(tt),
-            self._content._to_typetracer(forget_length),
+            self._content._to_typetracer(False),
             self._valid_when,
             unknown_length if forget_length else self.length,
             self._lsb_order,

diff --git a/src/awkward/contents/bytemaskedarray.py b/src/awkward/contents/bytemaskedarray.py
@@ -234,7 +234,7 @@ def _to_typetracer(self, forget_length: bool) -> Self:
         mask = self._mask.to_nplike(tt)
         return ByteMaskedArray(
             mask.forget_length() if forget_length else mask,
-            self._content._to_typetracer(forget_length),
+            self._content._to_typetracer(False),
             self._valid_when,
             parameters=self._parameters,
         )

diff --git a/src/awkward/contents/indexedarray.py b/src/awkward/contents/indexedarray.py
@@ -224,7 +224,7 @@ def _to_typetracer(self, forget_length: bool) -> Self:
         index = self._index.to_nplike(TypeTracer.instance())
         return IndexedArray(
             index.forget_length() if forget_length else index,
-            self._content._to_typetracer(forget_length),
+            self._content._to_typetracer(False),
             parameters=self._parameters,
         )
 

diff --git a/src/awkward/contents/indexedoptionarray.py b/src/awkward/contents/indexedoptionarray.py
@@ -213,7 +213,7 @@ def _to_typetracer(self, forget_length: bool) -> Self:
         index = self._index.to_nplike(TypeTracer.instance())
         return IndexedOptionArray(
             index.forget_length() if forget_length else index,
-            self._content._to_typetracer(forget_length),
+            self._content._to_typetracer(False),
             parameters=self._parameters,
         )
 

diff --git a/src/awkward/contents/listarray.py b/src/awkward/contents/listarray.py
@@ -228,7 +228,7 @@ def _to_typetracer(self, forget_length: bool) -> Self:
         return ListArray(
             starts.forget_length() if forget_length else starts,
             self._stops.to_nplike(tt),
-            self._content._to_typetracer(forget_length),
+            self._content._to_typetracer(False),
             parameters=self._parameters,
         )
 

diff --git a/src/awkward/contents/listoffsetarray.py b/src/awkward/contents/listoffsetarray.py
@@ -211,7 +211,7 @@ def _to_typetracer(self, forget_length: bool) -> Self:
         offsets = self._offsets.to_nplike(TypeTracer.instance())
         return ListOffsetArray(
             offsets.forget_length() if forget_length else offsets,
-            self._content._to_typetracer(forget_length),
+            self._content._to_typetracer(False),
             parameters=self._parameters,
         )
 

diff --git a/src/awkward/contents/unionarray.py b/src/awkward/contents/unionarray.py
@@ -486,7 +486,7 @@ def _to_typetracer(self, forget_length: bool) -> Self:
         return UnionArray(
             tags.forget_length() if forget_length else tags,
             self._index.to_nplike(tt),
-            [x._to_typetracer(forget_length) for x in self._contents],
+            [x._to_typetracer(False) for x in self._contents],
             parameters=self._parameters,
         )
 

diff --git a/tests/test_2765_slice_from_typetracer.py b/tests/test_2765_slice_from_typetracer.py