Update lgb.Booster.R

R-package/R/lgb.Booster.R

@@ -149,7 +149,7 @@ Booster <- R6::R6Class(
     add_valid = function(data, name) {
 
       # Check if data is lgb.Dataset
-      if (!lgb.check.r6.class(data, "lgb.Dataset")) {
+      if (!(lgb.Dataset = lgb.check.r6.class(data, "lgb.Dataset"))) {
         stop("lgb.Booster.add_valid: Can only use lgb.Dataset as validation data")
       }
 
@@ -216,7 +216,7 @@ Booster <- R6::R6Class(
       if (!is.null(train_set)) {
 
         # Check if training set is lgb.Dataset
-        if (!lgb.check.r6.class(train_set, "lgb.Dataset")) {
+        if (!(lgb.Dataset = lgb.check.r6.class(train_set, "lgb.Dataset"))) {
           stop("lgb.Booster.update: Only can use lgb.Dataset as training data")
         }
 
@@ -346,7 +346,7 @@ Booster <- R6::R6Class(
     eval = function(data, name, feval = NULL) {
 
       # Check if dataset is lgb.Dataset
-      if (!lgb.check.r6.class(data, "lgb.Dataset")) {
+      if (!(lgb.Dataset = lgb.check.r6.class(data, "lgb.Dataset"))) {
         stop("lgb.Booster.eval: Can only use lgb.Dataset to eval")
       }