bug fix

fix bugs in predict() and validate() when linked emulator was built with 'validate' mode.
mingdeyu · Nov 14, 2024 · 4184665 · 4184665
1 parent d3da50b
commit 4184665
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/R/prediction.R b/R/prediction.R
@@ -272,6 +272,9 @@ predict.lgp <- function(object, x, method = NULL, full_layer = FALSE, sample_siz
   }
   if ( !inherits(object,"lgp") ) stop("'object' must be an instance of the 'lgp' class.", call. = FALSE)
   if ( "metadata" %in% names(object$specs) ){
+    if ( !("emulator_obj" %in% names(object)) ){
+      stop("'object' is not in activation mode for predictions. Please set `mode = 'activate'` in `lgp()` to build the emulator.", call. = FALSE)
+    }
     if ( !is.matrix(x)&!is.vector(x) ) stop("'x' must be a vector or a matrix.", call. = FALSE)
     x <- unname(x)
     global_dim <- unique(subset(object$specs$struc, object$specs$struc[["From_Emulator"]] == "Global")$From_Output)

diff --git a/R/validation.R b/R/validation.R
@@ -617,6 +617,12 @@ validate.lgp <- function(object, x_test = NULL, y_test = NULL, method = NULL, sa
   }
   #check class
   if ( !inherits(object,"lgp") ) stop("'object' must be an instance of the 'lgp' class.", call. = FALSE)
+
+  if ( "metadata" %in% names(object$specs) ){
+    if ( !("emulator_obj" %in% names(object)) ){
+      stop("'object' is not in activation mode for validations. Please set `mode = 'activate'` in `lgp()` to build the emulator.", call. = FALSE)
+    }
+  }
   #check core number
   if( !is.null(cores) ) {
     cores <- as.integer(cores)