From 34baa3967988786d69713bb596f933bafb0b669f Mon Sep 17 00:00:00 2001 From: Daniel Date: Sat, 3 Aug 2024 20:08:47 +0200 Subject: [PATCH] add attr --- R/check_dag.R | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/R/check_dag.R b/R/check_dag.R index 43a26c5c2..945851bf2 100644 --- a/R/check_dag.R +++ b/R/check_dag.R @@ -150,13 +150,13 @@ check_dag.default <- function(..., dag <- .adjust_dag(dag, adjusted) } - .finalize_dag(dag, effect, outcome, exposure) + .finalize_dag(dag, effect, outcome, exposure, adjusted) } # helper ---------------------------------------------------------------------- -.finalize_dag <- function(dag, effect, outcome, exposure) { +.finalize_dag <- function(dag, effect, outcome, exposure, adjusted) { # data for checking effects checks <- lapply(c("direct", "total"), function(x) { adjustment_set <- unlist(dagitty::adjustmentSets(dag, effect = x), use.names = FALSE) @@ -172,6 +172,7 @@ check_dag.default <- function(..., attr(dag, "effect") <- effect attr(dag, "outcome") <- outcome attr(dag, "exposure") <- exposure + attr(dag, "adjusted") <- adjusted attr(dag, "check_direct") <- insight::compact_list(checks[[1]]) attr(dag, "check_total") <- insight::compact_list(checks[[2]])