Merge pull request #5 from danielmork/release

Hotfix for single cover model
danielmork · Mar 10, 2022 · a371a25 · a371a25
2 parents 2276e68 + 38382e2
commit a371a25
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/R/dlmtree.R b/R/dlmtree.R
@@ -376,7 +376,7 @@ dlmtree <- function(formula,
   if (model$intercept) {
     model$gamma[,1] <- model$gamma[,1] + model$Ymean
     if (ncol(model$Z) > 1)
-      model$gamma[,1] <- model$gamma[,1] - model$gamma[,-1] %*% model$Zmean[-1]
+      model$gamma[,1] <- model$gamma[,1] - model$gamma[,-1,drop=FALSE] %*% model$Zmean[-1]
   }
   colnames(model$gamma) <- model$Znames
 

diff --git a/R/tdlmm.R b/R/tdlmm.R
@@ -279,7 +279,7 @@ tdlmm <- function(formula,
   if (model$intercept) {
     model$gamma[,1] <- model$gamma[,1] + model$Ymean
     if (ncol(model$Z) > 1)
-      model$gamma[,1] <- model$gamma[,1] - model$gamma[,-1] %*% model$Zmean[-1]
+      model$gamma[,1] <- model$gamma[,1] - model$gamma[,-1,drop=FALSE] %*% model$Zmean[-1]
   }
   colnames(model$gamma) <- model$Znames
 

diff --git a/R/tdlnm.R b/R/tdlnm.R
@@ -356,7 +356,7 @@ tdlnm <- function(formula,
   if (model$intercept) {
     model$gamma[,1] <- model$gamma[,1] + model$Ymean
     if (ncol(model$Z) > 1)
-      model$gamma[,1] <- model$gamma[,1] - model$gamma[,-1] %*% model$Zmean[-1]
+      model$gamma[,1] <- model$gamma[,1] - model$gamma[,-1,drop=FALSE] %*% model$Zmean[-1]
   }
   colnames(model$gamma) <- model$Znames