small update

James-Thorson-NOAA · Apr 15, 2024 · b1c86bb · b1c86bb
1 parent 2d7274a
commit b1c86bb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/R/dsem.R b/R/dsem.R
@@ -873,9 +873,10 @@ logLik.dsem <- function(object, ...) {
 as_fitted_DAG <-
 function( fit,
           lag = 0,
-          what = "Estimate",
+          what = c("Estimate","Std_Error","p_value"),
           direction = 1 ){
 
+  what = match.arg(what)
   coefs = summary( fit )
   coefs = coefs[ which(coefs[,2]==lag), ]
   coefs = coefs[ which(coefs[,'direction'] %in% direction), ]