def to_runner_api(self, context): return beam_runner_api_pb2.StateSpec( set_spec=beam_runner_api_pb2.SetStateSpec( element_coder_id=context.coders.get_id(self.coder)))
def to_runner_api(self, context): # type: (PipelineContext) -> beam_runner_api_pb2.StateSpec return beam_runner_api_pb2.StateSpec( combining_spec=beam_runner_api_pb2.CombiningStateSpec( combine_fn=self.combine_fn.to_runner_api(context), accumulator_coder_id=context.coders.get_id(self.coder)))
def to_runner_api(self, context): # type: (PipelineContext) -> beam_runner_api_pb2.StateSpec return beam_runner_api_pb2.StateSpec( bag_spec=beam_runner_api_pb2.BagStateSpec( element_coder_id=context.coders.get_id(self.coder)))
def to_runner_api(self, context): # type: (PipelineContext) -> beam_runner_api_pb2.StateSpec return beam_runner_api_pb2.StateSpec( read_modify_write_spec=beam_runner_api_pb2. ReadModifyWriteStateSpec( coder_id=context.coders.get_id(self.coder)))