Skip to content

Commit

Permalink
check by id since no new objects should be generated here
Browse files Browse the repository at this point in the history
  • Loading branch information
lgray authored Dec 16, 2024
1 parent 08a88f7 commit aee6fd4
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/dask_awkward/lib/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -1295,7 +1295,7 @@ def _getitem_trivial_map_partitions(
else:
m = to_meta([where])[0]
meta = self._meta[m]
return _map_partitions(
return (
operator.getitem,
self,
where,
Expand All @@ -1315,7 +1315,7 @@ def _getitem_outer_bool_or_int_lazy_array(self, where):
)

new_meta = self._meta[where._meta]
return self._map_partitions(
return self.(
operator.getitem,
where,
meta=new_meta,
Expand All @@ -1335,7 +1335,7 @@ def _getitem_outer_str_or_list(
new_meta = self._meta[metad]
elif isinstance(where, (str, list)):
new_meta = self._meta[where]
return self._getitem_trivial_map_partitions(where, meta=new_meta, label=label)
return self._getitem_trivial(where, meta=new_meta, label=label)

def _getitem_outer_int(self, where: int | tuple[Any, ...]) -> Any:
if where == 0 or (isinstance(where, tuple) and where[0] == 0):
Expand Down Expand Up @@ -2164,7 +2164,7 @@ def map_partitions(
if len(kwargs) == 0:
non_traversed_deps, _ = unpack_collections(*args, traverse=False)
if all(
traversed_dep == non_traversed_dep
id(traversed_dep) == id(non_traversed_dep)
for traversed_dep, non_traversed_dep in zip(flat_deps, non_traversed_deps)
):
return _map_partitions(
Expand Down

0 comments on commit aee6fd4

Please sign in to comment.