tfx.dsl.compiler.compiler_utils.set_runtime_parameter_pb

View source on GitHub

Helper function to fill a RuntimeParameter proto.

View aliases

Main aliases

tfx.dsl.compiler.compiler.compiler_utils.set_runtime_parameter_pb

tfx.dsl.compiler.compiler_utils.set_runtime_parameter_pb(
    pb: pipeline_pb2.RuntimeParameter,
    name: Text,
    ptype: Type[tfx.types.Property],
    default_value: Optional[types.Property] = None
) -> pipeline_pb2.RuntimeParameter

Args
pb	A RuntimeParameter proto to be filled in.
name	Name to be set at pb.name.
ptype	The Python type to be set at pb.type.
default_value	Optional. If provided, it will be pb.default_value.

Returns
A RuntimeParameter proto filled with provided values.