Issue #911

vincentarelbundock · Sep 25, 2023 · b962b99 · b962b99
1 parent fffdefe
commit b962b99
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 1 deletion.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -1,6 +1,6 @@
 Package: marginaleffects
 Title: Predictions, Comparisons, Slopes, Marginal Means, and Hypothesis Tests
-Version: 0.15.1.001
+Version: 0.15.1.9002
 Authors@R: 
     c(person(given = "Vincent",
              family = "Arel-Bundock",

diff --git a/NEWS.md b/NEWS.md
@@ -3,6 +3,7 @@
 ## dev
 
 * Add support for `hypotheses()` to `inferences()`. This changes `FUN` argument name in `hypotheses` to `FUN` to not conflict with `inferences`. Thanks to @Tristan-Siegfried for code contribution #908.
+* Support `survival::survreg()`. Thanks to Carlisle Rainey for Report #911.
 
 ## 0.15.1
 

diff --git a/R/sanity_model.R b/R/sanity_model.R
@@ -98,6 +98,7 @@ sanity_model_supported_class <- function(model) {
         "speedglm",
         "speedlm",
         "stanreg",
+        "survreg",
         "svyolr",
         c("tobit", "survreg"),
         "tobit1",

diff --git a/R/type_dictionary.R b/R/type_dictionary.R
@@ -123,6 +123,9 @@ speedglm,response
 speedglm,link
 stanreg,response
 stanreg,link
+survreg,response
+survreg,link
+survreg,quantile
 svyglm,response
 svyglm,link
 svyolr,probs

diff --git a/data-raw/supported_models.csv b/data-raw/supported_models.csv
@@ -83,5 +83,6 @@ survey,svyglm,TRUE,TRUE,,,TRUE,TRUE,TRUE,TRUE
 survey,svyolr,TRUE,TRUE,,,,,,
 survival,clogit,TRUE,TRUE,,,,,,
 survival,coxph,TRUE,TRUE,TRUE,TRUE,U,U,TRUE,TRUE
+survival,survreg,TRUE,TRUE,,,,,,
 tobit1,tobit1,TRUE,TRUE,,,TRUE,TRUE,U,U
 truncreg,truncreg,TRUE,TRUE,TRUE,TRUE,TRUE,TRUE,U,U
diff --git a/inst/tinytest/test-pkg-survival.R b/inst/tinytest/test-pkg-survival.R
@@ -4,6 +4,13 @@ requiet("survival")
 requiet("emmeans")
 requiet("broom")
 
+
+# Issue #911: survreg support
+fit <- survreg(Surv(futime, fustat) ~ ecog.ps + rx, ovarian, dist='weibull', scale=1)
+s <- avg_slopes(fit)
+expect_inherits(s, "slopes")
+
+
 # clogit 
 N  <- 10000
 ng <- 5000
-Original file line number
+Diff line change
@@ Expand Up / @@ -3,6 +3,7 @@ @@
     ## dev
     * Add support for `hypotheses()` to `inferences()`. This changes `FUN` argument name in `hypotheses` to `FUN` to not conflict with `inferences`. Thanks to @Tristan-Siegfried for code contribution #908.
+    * Support `survival::survreg()`. Thanks to Carlisle Rainey for Report #911.
     ## 0.15.1
@@ Expand Down @@