Small 🐛 fix to write_output

iiasa · Feb 27, 2024 · d6c3c67 · d6c3c67
1 parent 2ceaa78
commit d6c3c67
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/R/class-biodiversityscenario.R b/R/class-biodiversityscenario.R
@@ -65,7 +65,7 @@ BiodiversityScenario <- R6::R6Class(
 
       message(paste0(
         ifelse(is.Waiver(self$limits),"Spatial-temporal scenario:","Spatial-temporal scenario (limited):"),
-        '\n  Used model: ',ifelse(is.Waiver(fit) || isFALSE(fit), text_red('None'), class(fit)[1] ),
+        '\n  Used model: ',ifelse(is.Waiver(fit) || isFALSE(fit), text_red('None'), fit$get_name() ),
         "\n --------- ",
         "\n  Predictors:     ", pn,
         "\n  Time period:    ", tp,

diff --git a/R/write_output.R b/R/write_output.R
@@ -57,9 +57,13 @@ methods::setMethod(
 
     # This function will only capture the distribution model object and will
     # save them separately
-    if(any(mod$get_name() %in% getOption("ibis.engines")) ){
-      # FIXME: If errors occur, check harmonization of saving among engines.
-      mod$save(fname = fname)
+    if(inherits(mod, "BiodiversityScenario")){
+      if(any(mod$get_name() %in% getOption("ibis.engines")) ){
+        # FIXME: If errors occur, check harmonization of saving among engines.
+        mod$save(fname = fname)
+      } else {
+        stop("Model engine not recognized? Older version?")
+      }
     } else if(is.Raster(mod)){
         if(tools::file_ext(fname) %in% c('tif', 'TIF')) {
           writeGeoTiff(file = mod, fname = fname, dt = dt)