Skip to content

Commit 884ce13

Browse files
committed
remove indirection in _ensure_cached method
1 parent a86f646 commit 884ce13

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

xarray/core/indexing.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -876,19 +876,15 @@ class MemoryCachedArray(ExplicitlyIndexedNDArrayMixin):
876876
def __init__(self, array):
877877
self.array = _wrap_numpy_scalars(as_indexable(array))
878878

879-
def _ensure_cached(self):
880-
self.array = as_indexable(self.array.get_duck_array())
881-
882-
async def _async_ensure_cached(self):
883-
duck_array = await self.array.async_get_duck_array()
884-
self.array = as_indexable(duck_array)
885-
886879
def get_duck_array(self):
887-
self._ensure_cached()
880+
# first ensure the array object is cached
881+
self.array = as_indexable(self.array.get_duck_array())
888882
return self.array.get_duck_array()
889883

890884
async def async_get_duck_array(self):
891-
await self._async_ensure_cached()
885+
# first ensure the array object is cached
886+
duck_array = await self.array.async_get_duck_array()
887+
self.array = as_indexable(duck_array)
892888
return await self.array.async_get_duck_array()
893889

894890
def _oindex_get(self, indexer: OuterIndexer):

0 commit comments

Comments
 (0)