def get(context):
try:
response = api.ml().models().get(name='projects/{}/models/{}'.format(
context.project_id, context.model_name)).execute()
return (response, None)
except errors.HttpError as err:
return (None, err)
def set_default(name):
try:
response = api.ml().models().versions().setDefault(name=name,
body={}).execute()
return (response, None)
except errors.HttpError as err:
return (None, err)
def create(context):
try:
response = api.ml().models().create(parent='projects/{}'.format(
context.project_id),
body={
'name': context.model_name
}).execute()
return (response, None)
except errors.HttpError as err:
return (None, err)
def list(context, **kwargs):
try:
response = api.ml().jobs().list(
parent='projects/{}'.format(context.project_id),
**kwargs
).execute()
pattern = re.compile(r'%s_\d{14}' % context.model_name)
return ([job for job in response['jobs'] if re.search(pattern, job['jobId'])], None)
except errors.HttpError as err:
return (None, err)
def create(context, version):
try:
response = api.ml().models().versions().create(
parent='projects/{}/models/{}'.format(context.project_id,
context.model_name),
body={
'name': 'v' + version,
'deploymentUri': context.deployment_uri(version)
}).execute()
return (response, None)
except errors.HttpError as err:
return (None, err)
def submit(packages, context):
args = context.args
body = {
'jobId': context.job_id,
'trainingInput': {
'packageUris': packages,
'pythonModule': args.module_name,
'region': context.region,
'args': context.train_args()
}
}
try:
response = api.ml().jobs().create(
parent='projects/{}'.format(context.project_id),
body=body
).execute()
return (response, None)
except errors.HttpError as err:
return (None, err)