apache_beam.runners.direct.helper_transforms module

class apache_beam.runners.direct.helper_transforms.LiftedCombinePerKey(combine_fn, args, kwargs)[source]

Bases: apache_beam.transforms.ptransform.PTransform

An implementation of CombinePerKey that does mapper-side pre-combining.

expand(pcoll)[source]
class apache_beam.runners.direct.helper_transforms.PartialGroupByKeyCombiningValues(combine_fn)[source]

Bases: apache_beam.transforms.core.DoFn

Aggregates values into a per-key-window cache.

As bundles are in-memory-sized, we don’t bother flushing until the very end.

start_bundle()[source]
process(element, window=WindowParam)[source]
finish_bundle()[source]
default_type_hints()[source]
class apache_beam.runners.direct.helper_transforms.FinishCombine(combine_fn)[source]

Bases: apache_beam.transforms.core.DoFn

Merges partially combined results.

process(element)[source]
default_type_hints()[source]