schrodinger.stepper.shims module

Module providing shims that convert stepper steps into Beam transforms.

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

Bases: apache_beam.transforms.ptransform.PTransform

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

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

expand(pcoll)

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

Bases: apache_beam.transforms.ptransform.PTransform

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

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

expand(pcoll)