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())