fixed automl cleanup from R client

h2oai · Oct 3, 2019 · a453042 · a453042
1 parent 8a9a9f4
commit a453042
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/h2o-r/h2o-package/R/automl.R b/h2o-r/h2o-package/R/automl.R
@@ -285,6 +285,7 @@ h2o.automl <- function(x, y, training_frame,
 
   # GET AutoML object
   aml <- h2o.getAutoML(project_name = res$job$dest$name)
+  attr(aml, "id") <- res$job$dest$name
   return(aml)
 }
 

diff --git a/h2o-r/h2o-package/R/classes.R b/h2o-r/h2o-package/R/classes.R
@@ -869,4 +869,4 @@ setClass("H2OAutoML", slots = c(project_name = "character",
                                 training_info = "list"),
                       contains = "Keyed")
 #' @rdname h2o.keyof
-setMethod("h2o.keyof", signature("H2OAutoML"), function(object) object@project_name)
+setMethod("h2o.keyof", signature("H2OAutoML"), function(object) attr(object, "id"))