apache_beam.typehints.typecheck module

Runtime type checking support.

For internal use only; no backwards-compatibility guarantees.

class apache_beam.typehints.typecheck.AbstractDoFnWrapper(dofn)

Bases: apache_beam.transforms.core.DoFn

An abstract class to create wrapper around DoFn

wrapper(method, args, kwargs)

start_bundle(*args, **kwargs)

process(*args, **kwargs)

finish_bundle(*args, **kwargs)

class apache_beam.typehints.typecheck.OutputCheckWrapperDoFn(dofn, full_label)

Bases: apache_beam.typehints.typecheck.AbstractDoFnWrapper

A DoFn that verifies against common errors in the output type.

wrapper(method, args, kwargs)

class apache_beam.typehints.typecheck.TypeCheckWrapperDoFn(dofn, type_hints, label=None)

Bases: apache_beam.typehints.typecheck.AbstractDoFnWrapper

A wrapper around a DoFn which performs type-checking of input and output.

wrapper(method, args, kwargs)

process(*args, **kwargs)

class apache_beam.typehints.typecheck.TypeCheckCombineFn(combinefn, type_hints, label=None)

Bases: apache_beam.transforms.core.CombineFn

A wrapper around a CombineFn performing type-checking of input and output.

create_accumulator(*args, **kwargs)

add_input(accumulator, element, *args, **kwargs)

merge_accumulators(accumulators, *args, **kwargs)

compact(accumulator, *args, **kwargs)

extract_output(accumulator, *args, **kwargs)

class apache_beam.typehints.typecheck.TypeCheckVisitor

Bases: apache_beam.pipeline.PipelineVisitor

enter_composite_transform(applied_transform)

leave_composite_transform(applied_transform)

visit_transform(applied_transform)