def get_trial(stub):
try:
reply = stub.GetTrial(api_pb2.GetTrialRequest(trial_name=TEST_TRIAL), 10)
trial = reply.trial
if trial and trial.name == TEST_TRIAL:
logger.info("Get trial %s successfully" % TEST_TRIAL)
else:
raise Exception()
except:
logger.error("Failed to get trial %s" % TEST_TRIAL, exc_info=True)
raise
def get_best_trial(trial_id):
vizier_core = "vizier-core.kubeflow:6789"
with grpc.insecure_channel(vizier_core) as channel:
stub = api_pb2_grpc.ManagerStub(channel)
response = stub.GetTrial(api_pb2.GetTrialRequest(trial_id=trial_id))
return response.trial