diff --git a/src/awkward/contents/bitmaskedarray.py b/src/awkward/contents/bitmaskedarray.py index b5c1370efd..ae5e8e2fe5 100644 --- 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 index 1ba6f8966d..ab82e40764 100644 --- 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 index eb5f595337..d990dea323 100644 --- 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 index cee5e77326..f1d9568a55 100644 --- 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 index 26f872f421..98f153925b 100644 --- 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 index d3efb08698..c6aa5f1d8f 100644 --- 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 index 95ddfca6e4..2d0647ce77 100644 --- 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 deleted file mode 100644 index b144e7f9be..0000000000 --- a/tests/test_2765_slice_from_typetracer.py +++ /dev/null @@ -1,15 +0,0 @@ -import awkward as ak - - -def test(): - array = ak.typetracer.typetracer_from_form( - ak.forms.from_type( - ak.types.from_datashape("{x: var * {y:int64}}", highlevel=False) - ) - ) - sliced = array[0, "x", 0] - assert sliced.type.is_equal_to( - ak.types.ScalarType( - ak.types.RecordType([ak.types.NumpyType("int64")], ["y"]), None - ) - )