R/delete_model.R

In jcpsantiago/sagemaker-multimodel-R: Sagemaker Multi Model API

#' Unload loaded model
#'
#' @param model_name Name of the model
#' @param req Request object
#' @param res Response object
#'
#' @return Unloads a loaded model
#' @export
delete_model <- function(model_name, req, res) {
  if (!model_name %in% .sage$loaded_models$modelName) {
    message("Model `", model_name, "` is not loaded!")
    res$status <- 404
    return("")
  }

  message("Unloading model `", model_name, "`")
  .sage$loaded_models <- subset(
    .sage$loaded_models, modelName != model_name,
    drop = FALSE
  )
  rm(list = eval(model_name), envir = .sage)

  return("")
}