Another aggregateDesigns() ordering adjustment

On second thought, what was intended was something like this. Cf. #138
markmfredrickson · Nov 8, 2024 · 42b9004 · 42b9004
1 parent 01cbab8
commit 42b9004
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))