Export extractAIC for CoxPH and add more internal functions

h2oai · Nov 6, 2023 · 78beec4 · 78beec4
1 parent 0d2c340
commit 78beec4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/h2o-r/h2o-package/R/classes.R b/h2o-r/h2o-package/R/classes.R
@@ -451,6 +451,7 @@ coef.H2OCoxPHModelSummary <- function(object, ...) object@summary$coefficients
 #' @param fit an \code{H2OCoxPHModel} object.
 #' @param scale optional numeric specifying the scale parameter of the model.
 #' @param k numeric specifying the weight of the equivalent degrees of freedom.
+#' @export
 extractAIC.H2OCoxPHModel <- function(fit, scale, k = 2, ...) {
   fun <- get("extractAIC.coxph", getNamespace("stats"))
   if (missing(scale))

diff --git a/h2o-r/scripts/h2o-r-test-setup.R b/h2o-r/scripts/h2o-r-test-setup.R
@@ -197,7 +197,8 @@ function() {
             "h2o.logIt", ".h2o.__remoteSend", ".as.survival.coxph.model", ".as.survival.coxph.summary",
             ".getExpanded", ".str.list", "is.H2OFrame", ".get.session.property", ".set.session.property",
             ".h2o.maximizing_metrics", ".h2o.doSafeGET", ".parse.h2oconfig", ".h2o.check_java_version",
-            "cut.H2OFrame", "as.data.frame.H2OFrame", ".h2o.perfect_auc"
+            "cut.H2OFrame", "as.data.frame.H2OFrame", ".h2o.perfect_auc", ".newExpr", ".h2o.doSafeREST",
+            ".h2o.fromJSON"
          )
 
         for (fn in additional_imports)