#NEWLINE# def fetch_watermark_hold( self , state_key ) : #NEWLINE# #INDENT# 'Get the watermark hold at given state tag.' #NEWLINE# request = windmill_pb2 . GetDataRequest ( ) #NEWLINE# computation_request = windmill_pb2 . ComputationGetDataRequest ( computation_id = self . computation_id ) #NEWLINE# keyed_request = windmill_pb2 . KeyedGetDataRequest ( key = self . key , work_token = self . work_token ) #NEWLINE# keyed_request . watermark_holds_to_fetch . add ( tag = state_key , state_family = '' ) #NEWLINE# computation_request . requests . extend ( [ keyed_request ] ) #NEWLINE# request . requests . extend ( [ computation_request ] ) #NEWLINE# return self . windmill . GetData ( request )