Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix #4764 (#4800)

Browse Source 
check all classes in xgb.get.handle
This commit is contained in:
Michaël Benesty 2019-08-22 03:42:37 +02:00 committed by Tong He
parent 6e6216ad67
commit 41227d1933
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
R-package/R/xgb.Booster.R
View File

@ -51,11 +51,13 @@ is.null.handle <- function(handle) {
# Return a verified to be valid handle out of either xgb.Booster.handle or xgb.Booster # Return a verified to be valid handle out of either xgb.Booster.handle or xgb.Booster
# internal utility function # internal utility function
xgb.get.handle <- function(object) { xgb.get.handle <- function(object) {
handle <- switch(class(object)[1], if (inherits(object, "xgb.Booster")) {
xgb.Booster = object$handle, handle <- object$handle
xgb.Booster.handle = object, } else if (inherits(object, "xgb.Booster.handle")) {
handle <- object
} else {
stop("argument must be of either xgb.Booster or xgb.Booster.handle class") stop("argument must be of either xgb.Booster or xgb.Booster.handle class")
) }
if (is.null.handle(handle)) { if (is.null.handle(handle)) {
stop("invalid xgb.Booster.handle") stop("invalid xgb.Booster.handle")
} }