Fix R load raw.

R-package/R/xgb.load.raw.R

@@ -5,10 +5,18 @@
 #' @param buffer the buffer returned by xgb.save.raw
 #'
 #' @export
-xgb.load.raw <- function(buffer) {
+xgb.load.raw <- function(buffer, as_booster = TRUE) {
   cachelist <- list()
   handle <- .Call(XGBoosterCreate_R, cachelist)
   .Call(XGBoosterLoadModelFromRaw_R, handle, buffer)
   class(handle) <- "xgb.Booster.handle"
-  return (handle)
+
+  if (as_booster) {
+    booster <- list(handle = handle, raw = NULL)
+    class(booster) <- "xgb.Booster"
+    booster <- xgb.Booster.complete(booster, saveraw = TRUE)
+    return(booster)
+  } else {
+    return (handle)
+  }
 }

R-package/tests/testthat/test_io.R

@@ -22,15 +22,6 @@ test_that("load/save raw works", {
   from_json <- xgb.load.raw(json_bytes)
   from_ubj <- xgb.load.raw(ubj_bytes)
 
-  ## FIXME(jiamingy): Should we include these 3 lines into `xgb.load.raw`?
-  from_json <- list(handle = from_json, raw = NULL)
-  class(from_json) <- "xgb.Booster"
-  from_json <- xgb.Booster.complete(from_json, saveraw = TRUE)
-
-  from_ubj <- list(handle = from_ubj, raw = NULL)
-  class(from_ubj) <- "xgb.Booster"
-  from_ubj <- xgb.Booster.complete(from_ubj, saveraw = TRUE)
-
   json2old <- xgb.save.raw(from_json, raw_format = "deprecated")
   ubj2old <- xgb.save.raw(from_ubj, raw_format = "deprecated")