From 47b5cf5148e5eb933e5a798a8a3dc20b87378844 Mon Sep 17 00:00:00 2001
From: hetong007 <hetong007@gmail.com>
Date: Mon, 9 Feb 2015 17:35:50 -0800
Subject: [PATCH] fix save.raw

---
 R-package/R/xgb.save.raw.R | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/R-package/R/xgb.save.raw.R b/R-package/R/xgb.save.raw.R
index 91f7075bd..ee217aa85 100644
--- a/R-package/R/xgb.save.raw.R
+++ b/R-package/R/xgb.save.raw.R
@@ -18,6 +18,9 @@
 #' @export
 #' 
 xgb.save.raw <- function(handle) {
+  if (class(handle) == "xgb.Booster"){
+    handle <- handle$handle
+  }
   if (class(handle) == "xgb.Booster.handle") {
     raw <- .Call("XGBoosterModelToRaw_R", handle, PACKAGE = "xgboost")
     return(raw)