schrodinger.stepper.shims module

Module providing shims that convert stepper steps into Beam transforms.

class schrodinger.stepper.shims.ShimMapStep(step: Type[MapStep], **kwargs)

Bases: PTransform

A transform that takes a stepper.MapStep class and its settings and yields its outputs.

__init__(step: Type[MapStep], **kwargs)

expand(pcoll)

class schrodinger.stepper.shims.ShimReduceStep(step: Type[ReduceStep], **kwargs)

Bases: PTransform

A transform that takes a stepper.ReduceStep class and its settings and yields its outputs.

__init__(step: Type[ReduceStep], **kwargs)

expand(pcoll)