small bug fix in variance calculation for t

nicholasjclark · Mar 24, 2024 · 5698279 · 5698279
1 parent 7e041a5
commit 5698279
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/R/compute_edf.R b/R/compute_edf.R
@@ -79,7 +79,10 @@ compute_edf = function(mgcv_model, object, rho_names, sigma_raw_names){
     # Calculate variance using family's mean-variance relationship
     mu_variance <- predict(object,
                            process_error = FALSE,
-                           type = 'variance')[best_draw, 1:length(eta)]
+                           type = 'variance')[best_draw, ]
+    if(length(mu_variance) > 1){
+      mu_variance <- mu_variance[1:length(eta)]
+    }
     if(any(mu_variance == 0)){
       mu_variance[which(mu_variance == 0)] <-
         mu[which(mu_variance == 0)]

diff --git a/src/mvgam.dll b/src/mvgam.dll