MCPcopy
hub / github.com/dask/dask / _layer

Method _layer

dask/dataframe/dask_expr/_reductions.py:342–370  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

340 raise ValueError("split_every must be greater than 1 or False")
341
342 def _layer(self):
343 # apply combine to batches of intermediate results
344 j = 1
345 d = {}
346 keys = self.frame.__dask_keys__()
347 split_every = self.split_every
348 while split_every is not False and len(keys) > split_every:
349 new_keys = []
350 for i, batch in enumerate(
351 toolz.partition_all(split_every or len(keys), keys)
352 ):
353 batch = list(batch)
354 if self.combine_kwargs:
355 d[self._name, j, i] = (
356 apply,
357 self.combine,
358 [batch],
359 self.combine_kwargs,
360 )
361 else:
362 d[self._name, j, i] = (self.combine, batch)
363 new_keys.append((self._name, j, i))
364 j += 1
365 keys = new_keys
366
367 # apply aggregate to the final result
368 d[self._name, 0] = (apply, self.aggregate, [keys], self.aggregate_kwargs)
369
370 return d
371
372 @property
373 def _meta(self):

Callers

nothing calls this directly

Calls 1

__dask_keys__Method · 0.45

Tested by

no test coverage detected