From 2efab45f7339dc3bd9b1fa04347f2f76ed5d4c18 Mon Sep 17 00:00:00 2001 From: Iason Krommydas Date: Thu, 6 Feb 2025 20:59:25 -0600 Subject: [PATCH] whoops --- src/awkward/_nplikes/virtual.py | 8 ++++---- src/awkward/operations/ak_dematerialize.py | 2 +- src/awkward/operations/ak_materialize.py | 4 ++-- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/src/awkward/_nplikes/virtual.py b/src/awkward/_nplikes/virtual.py index 2a4c0b734e..c730ff86e0 100644 --- a/src/awkward/_nplikes/virtual.py +++ b/src/awkward/_nplikes/virtual.py @@ -30,12 +30,12 @@ def materialize_if_virtual(*args: Any) -> tuple[Any, ...]: ) -def unmaterialize_if_virtual(*args: Any) -> tuple[Any, ...]: +def dematerialize_if_virtual(*args: Any) -> tuple[Any, ...]: """ - A little helper function to unmaterialize all virtual arrays in a list of arrays. + A little helper function to dematerialize all virtual arrays in a list of arrays. """ return tuple( - arg.unmaterialize() if isinstance(arg, VirtualArray) else arg for arg in args + arg.dematerialize() if isinstance(arg, VirtualArray) else arg for arg in args ) @@ -106,7 +106,7 @@ def materialize(self) -> ArrayLike: self._array = self._nplike.asarray(self.generator()) return cast(ArrayLike, self._array) - def unmaterialize(self) -> VirtualArray: + def dematerialize(self) -> VirtualArray: if self._array is not UNMATERIALIZED: self._materialized_form_keys.remove(self.form_key) self._array = UNMATERIALIZED diff --git a/src/awkward/operations/ak_dematerialize.py b/src/awkward/operations/ak_dematerialize.py index 5a9681e692..82ee41664d 100644 --- a/src/awkward/operations/ak_dematerialize.py +++ b/src/awkward/operations/ak_dematerialize.py @@ -35,7 +35,7 @@ def action(layout, backend, **kwargs): if isinstance(layout, ak.contents.NumpyArray): buffer = layout.data if isinstance(buffer, VirtualArray): - out = buffer.unmaterialize() + out = buffer.dematerialize() else: out = VirtualArray( nplike_of_obj(buffer), buffer.shape, buffer.dtype, lambda: buffer diff --git a/src/awkward/operations/ak_materialize.py b/src/awkward/operations/ak_materialize.py index e995cefe2f..ee5de9cf80 100644 --- a/src/awkward/operations/ak_materialize.py +++ b/src/awkward/operations/ak_materialize.py @@ -23,10 +23,10 @@ def materialize( yield (array,) # Implementation - return _impl1(array, highlevel, behavior, attrs) + return _impl(array, highlevel, behavior, attrs) -def _impl1(array, highlevel, behavior, attrs): +def _impl(array, highlevel, behavior, attrs): with HighLevelContext(behavior=behavior, attrs=attrs) as ctx: layout = ctx.unwrap(array, allow_record=True, primitive_policy="error")