restore accidental deletions

dask-contrib · Dec 17, 2024 · 7fd8890 · 7fd8890
1 parent 9118b9e
commit 7fd8890
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/dask_awkward/lib/core.py b/src/dask_awkward/lib/core.py
@@ -1302,7 +1302,7 @@ def _getitem_trivial_map_partitions(
             else:
                 m = to_meta([where])[0]
                 meta = self._meta[m]
-        return (
+        return _map_partitions(
             operator.getitem,
             self,
             where,
@@ -1322,7 +1322,7 @@ def _getitem_outer_bool_or_int_lazy_array(self, where):
             )
 
         new_meta = self._meta[where._meta]
-        return self.(
+        return self._map_partitions(
             operator.getitem,
             where,
             meta=new_meta,
@@ -1342,7 +1342,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(where, meta=new_meta, label=label)
+        return self._getitem_trivial_map_partitions(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):