get var as diag

lukesonnet · Nov 16, 2018 · 4239e16 · 4239e16
1 parent 5363a35
commit 4239e16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/inference.R b/R/inference.R
@@ -230,7 +230,7 @@ inference.krls2 <- function(obj,
         if (!binaryindicator[i]) {
           deriv_list <- pwmfx(obj$K, X[, i, drop = FALSE], obj$coeffs, vcov.c, tau, tau2, obj$b)
           derivatives[, i] <- deriv_list$deriv
-          var.derivatives[, i] <- deriv_list$var_deriv
+          var.derivatives[, i] <- diag(deriv_list$var_deriv)
           var.avgderivatives[i] <- deriv_list$var_avg_deriv
         }
       }