same as prev. commit but coding is more elegant

ycroissant · Dec 27, 2024 · ddbe1c8 · ddbe1c8
1 parent fe8baa0
commit ddbe1c8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/R/est_gmm.R b/R/est_gmm.R
@@ -710,8 +710,9 @@ extract.data <- function(data, as.matrix = TRUE){
     X <- cbind(data[[1L]], X)
     colnames(X)[1L] <- deparse(trms[[2L]])
   }
-  data <- if(!as.matrix) collapse::rsplit(as.data.frame(X), index[[1L]], simplify = FALSE)
-          else           collapse::rsplit(X, index[[1L]], simplify = FALSE)
+
+  if(!as.matrix) X <- as.data.frame(X)
+  data <- collapse::rsplit(X, index[[1L]], simplify = FALSE)
   time <- collapse::gsplit(index[[2L]], index[[1L]])
   data <- mapply(
                  function(x, y){