def testPatchNoUpdate(self):
model_ref = resources.REGISTRY.Parse('myModel',
params={'projectsId': self.Project()},
collection='ml.projects.models')
no_label_update = labels_util.UpdateResult(False, None)
with self.assertRaises(models.NoFieldsSpecifiedError):
self.models_client.Patch(model_ref, no_label_update)
def testPatchDescription(self):
updated_description = 'My New Description'
model_ref = resources.REGISTRY.Parse('myModel',
params={'projectsId': self.Project()},
collection='ml.projects.models')
model = self.short_msgs.Model(description=updated_description)
op = self.msgs.GoogleLongrunningOperation(name='my-op')
self.client.projects_models.Patch.Expect(
self._MakePatchRequest(model_ref=model_ref, model=model,
update_mask=['description']), op)
no_label_update = labels_util.UpdateResult(False, None)
self.assertEqual(op, self.models_client.Patch(
model_ref, no_label_update, updated_description))
def testPatchLabels(self):
labels = {'foo': 'bar', 'fizz': 'buzz'}
labels_field = encoding.DictToAdditionalPropertyMessage(
labels, self.short_msgs.Model.LabelsValue, True)
model_ref = resources.REGISTRY.Parse('myModel',
params={'projectsId': self.Project()},
collection='ml.projects.models')
model = self.short_msgs.Model(labels=labels_field)
op = self.msgs.GoogleLongrunningOperation(name='my-op')
self.client.projects_models.Patch.Expect(
self._MakePatchRequest(model_ref=model_ref, model=model,
update_mask=['labels']), op)
label_update = labels_util.UpdateResult(True, labels_field)
self.assertEqual(op, self.models_client.Patch(model_ref, label_update))
def testPatch(self):
labels = {'foo': 'bar', 'fizz': 'buzz'}
updated_description = 'My New Description'
labels_field = self.short_msgs.Version.LabelsValue(
additionalProperties=[
self.short_msgs.Version.LabelsValue.AdditionalProperty(
key=key, value=value)
for key, value in sorted(labels.items())
])
version = self.short_msgs.Version(labels=labels_field,
description=updated_description)
op = self.msgs.GoogleLongrunningOperation(name='my-op')
self.client.projects_models_versions.Patch.Expect(
self._MakePatchRequest(self.version_ref,
version,
update_mask=['description', 'labels']), op)
label_update = labels_util.UpdateResult(True, labels_field)
self.assertEqual(
op,
self.versions.Patch(self.version_ref, label_update,
updated_description))
def testPatchNoUpdate(self):
no_label_update = labels_util.UpdateResult(False, None)
with self.assertRaises(versions_api.NoFieldsSpecifiedError):
self.versions.Patch(self.version_ref, no_label_update)