def test_delete_registered_model_tag(mock_get_request_message, mock_model_registry_store):
name = "model1"
key = "some weird key"
mock_get_request_message.return_value = DeleteRegisteredModelTag(name=name, key=key)
_delete_registered_model_tag()
_, args = mock_model_registry_store.delete_registered_model_tag.call_args
assert args == {"name": name, "key": key}
def test_delete_registered_model_tag(self, mock_http):
name = "model_1"
self.store.delete_registered_model_tag(name=name, key="key")
self._verify_requests(
mock_http,
"registered-models/delete-tag",
"DELETE",
DeleteRegisteredModelTag(name=name, key="key"),
)
def delete_registered_model_tag(self, name, key):
"""
Delete a tag associated with the registered model.
:param name: Registered model name.
:param key: Registered model tag key.
:return: None
"""
req_body = message_to_json(DeleteRegisteredModelTag(name=name, key=key))
self._call_endpoint(DeleteRegisteredModelTag, req_body)
def _delete_registered_model_tag():
request_message = _get_request_message(DeleteRegisteredModelTag())
_get_model_registry_store().delete_registered_model_tag(
name=request_message.name,
key=request_message.key)
return _wrap_response(DeleteRegisteredModelTag.Response())
