modify list structure of inits for vb()

ecmerkle · Aug 28, 2023 · 48af49d · 48af49d
1 parent 687d4bf
commit 48af49d
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -1,6 +1,6 @@
 Package: blavaan
 Title: Bayesian Latent Variable Analysis
-Version: 0.4-9.1158
+Version: 0.4-9.1159
 Authors@R: c(person(given = "Edgar", family = "Merkle",
                     role = c("aut", "cre"),
                     email = "merklee@missouri.edu",

diff --git a/R/blav_object_methods.R b/R/blav_object_methods.R
@@ -127,8 +127,13 @@ function(object, header       = TRUE,
         }
         PE$group[PE$group == 0] <- 1
 
-        attributes(PE)$information <- "MCMC"
-        attributes(PE)$se <- "MCMC"
+        if(object@call$target == "vb") {
+            attributes(PE)$information <- "VB"
+            attributes(PE)$se <- "VB"
+        } else {
+            attributes(PE)$information <- "MCMC"
+            attributes(PE)$se <- "MCMC"
+        }
         ## 95% HPD; FIXME display blanks for equality-constrained parameters
         ##                (like Std.Err column)
 

diff --git a/R/blavaan.R b/R/blavaan.R
@@ -712,6 +712,7 @@ blavaan <- function(...,  # default lavaan arguments
                     rjcall <- "stan"
                 } else if(usevb){
                     rjcall <- "vb"
+                    rjarg$init <- rjarg$init[[1]]
                 } else if(target == "cmdstan"){
                     fname <- paste0("stanmarg_", packageDescription("blavaan")["Version"])
                     fdir <- paste0(cmdstanr::cmdstan_path(), "/")