R/assertions.R

In mlr-org/mlr3forecasting: Forecasting for 'mlr3'

# assertion to ensure a helpful error message
assert_prediction_count = function(actual, expected, type) {
  if (actual != expected) {
    if (actual < expected) {
      stopf("Predicted %s not complete, %s for %i observations is missing",
        type, type, expected - actual)
    } else {
      stopf("Predicted %s contains %i additional predictions without matching rows",
        type, actual - expected)
    }
  }
}

nullify_nulldt = function(x) {
  if (nrow(x) == 0) {
    return(NULL)
  }
  return(x)
}

new_prediction_data = function(li, task_type = NULL) {
  li = discard(li, is.null)
  class(li) = sprintf("PredictionData%s", c(capitalize(task_type), ""))
  li
}