Merge pull request #140 from markmfredrickson/i138-aggD

Another aggregateDesigns() ordering adjustment
markmfredrickson · Nov 13, 2024 · df13b68 · df13b68
2 parents 23ab550 + 42b9004
commit df13b68
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/R/Design.R b/R/Design.R
@@ -694,10 +694,10 @@ aggregateDesigns <- function(design) {
   # To align w/ this `C_transp`, everything to be returned 
   # needs to align w/ `levels(Cluster)`, not with 
   # `Cluster` itself. So,
-  Z <- Z[levels(Cluster)]
-  StrataFrame  <-
-      StrataFrame[match(levels(Cluster), as.character(Cluster)),
-                  , drop=FALSE]
+  cperm <- match(levels(Cluster), as.character(Cluster))
+  Z <- Z[cperm]
+  StrataFrame  <- StrataFrame[cperm, , drop=FALSE]
+  Cluster <- Cluster[cperm]
 
   unit.weights <- as.matrix(C_transp %*% as.matrix(design@UnitWeights))