return id-constant probs only once

helske · Dec 5, 2024 · d0493a6 · d0493a6
1 parent 2240195
commit d0493a6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/R/get_probs.R b/R/get_probs.R
@@ -34,6 +34,7 @@ get_initial_probs.nhmm <- function(model, probs, ...) {
   }
   if (attr(model$X_pi, "icpt_only")) {
     X <- model$X_pi[, 1L, drop = FALSE]
+    ids <- "all"
   } else {
     X <- model$X_pi
   }
@@ -104,6 +105,7 @@ get_transition_probs.nhmm <- function(model, probs, remove_voids = TRUE, ...) {
   }
   if (!attr(model$X_A, "iv")) {
     X <- model$X_A[, , 1L, drop = FALSE]
+    ids <- "all"
   } else {
     X <- model$X_A
   }
@@ -192,6 +194,7 @@ get_emission_probs.nhmm <- function(model, probs, remove_voids = TRUE, ...) {
   }
   if (!attr(model$X_B, "iv")) {
     X <- model$X_B[, , 1L, drop = FALSE]
+    ids <- "all"
   } else {
     X <- model$X_B
   }
@@ -286,6 +289,7 @@ get_cluster_probs.mnhmm <- function(model, probs, ...) {
   }
   if (attr(model$X_omega, "icpt_only")) {
     X <- model$X_omega[, 1L, drop = FALSE]
+    ids <- "all"
   } else {
     X <- model$X_omega
   }