R/download_desc_run.R

In mlr-org/mlr3oml: Connector Between 'mlr3' and 'OpenML'

download_desc_run = function(run_id, server) {
  desc = get_json(paste0(server, "/json/run/%i"), run_id, server = server)[[1L]]
  desc = parse_desc_run(desc)
  return(desc)
}

parse_desc_run = function(desc) {
  # OpenML response does not return correct types.
  desc$run_id = as.integer(desc$run_id)
  desc$input_data$dataset$did = as.integer(desc$input_data$dataset$did)
  desc$uploader = as.integer(desc$uploader)
  desc$task_id = as.integer(desc$task_id)
  desc$flow_id = as.integer(desc$flow_id)
  desc$setup_id = as.integer(desc$flow_id)
  desc$output_data$file = as.data.table(desc$output_data$file)
  desc$parameter_setting = as.data.table(desc$parameter_setting)

  # evaluation: | name | value | array_data | repeat | fold |
  if (is.null(desc$output_data$evaluation)) {
    desc$output_data$evaluation = data.table(
      name = character(0),
      value = numeric(0),
      array_data = list(),
      repeat. = integer(0),
      fold = integer(0)
    )
  } else {
    desc$output_data$evaluation = as.data.table(desc$output_data$evaluation)

    # The contents of the file are stored as strings again that correspond (mostly) to json
    # format and therefore have to be parsed again
    if ("array_data" %in% colnames(desc$output_data$evaluation)) {
      desc$output_data$evaluation[, `:=`("array_data", map(get("array_data"), .f = parse_json_safely))] # nolint
    }
    desc$output_data$evaluation[, `:=`("value", as.numeric(get("value")))]
    if ("repeat." %in% colnames(desc$output_data$evaluation)) {
      desc$output_data$evaluation[, `:=`("repeat.", as.integer(get("repeat.")))] # nolint
    }
    if ("fold" %in% colnames(desc$output_data$evaluation)) {
      desc$output_data$evaluation[, `:=`("fold", as.integer(get("fold")))] # nolint
    }
  }
  return(desc)
}

parse_json_safely = function(x) {
  x = tryCatch(jsonlite::fromJSON(x),
    error = function(cond) {
      x
    }
  )
}