From fb54e23dafd0465f88dbcd19a2d6a30820a208f4 Mon Sep 17 00:00:00 2001
From: Hannah Frick <hannah@rstudio.com>
Date: Wed, 11 May 2022 17:46:08 +0100
Subject: [PATCH] keep dimensions for single row in `newdata`

---
 R/survival.R | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/R/survival.R b/R/survival.R
index 071eed4..26a7750 100644
--- a/R/survival.R
+++ b/R/survival.R
@@ -29,7 +29,7 @@ survFit.mboost <- function(object, newdata = NULL, ...)
     devent <- n.event > 0
     if (!is.null(newdata)){
         S <- exp( tcrossprod( -H, exp(as.numeric(predict(object,
-        newdata=newdata)- mean(predict(object)))) ))[devent,]
+        newdata=newdata)- mean(predict(object)))) ))[devent, , drop = FALSE]
         colnames(S) <- rownames(newdata)
     } else S <- matrix(exp(-H)[devent], ncol = 1)