Fix nesmr AIC

jean997 · Oct 28, 2024 · 88b5cc4 · 88b5cc4
1 parent 28db382
commit 88b5cc4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/aic.R b/R/aic.R
@@ -1,6 +1,6 @@
 #' @export
 nesmr_aic <- function(fit, ...) {
-  if (fit$is_nesmr) {
-    -2 * log_py(fit, ...) + 2 * sum(fit$direct_effect_template)
+  if ("B_template" %in% names(fit)) {
+    -2 * log_py(fit, ...) + 2 * sum(fit$B_template)
   }
 }