added drop = FALSE ton handle the case when Q = 1 in reordering

GrossSBM · Apr 29, 2021 · 8c50c3c · 8c50c3c
1 parent 42658e2
commit 8c50c3c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/R/R6Class-BipartiteSBM_fit.R b/R/R6Class-BipartiteSBM_fit.R
@@ -136,7 +136,7 @@ BipartiteSBM_fit <-
         oCol <- order(private$pi[[1]] %*% private$theta$mean, decreasing = TRUE)
         private$pi[[1]] <- private$pi[[1]][oRow]
         private$pi[[2]] <- private$pi[[2]][oCol]
-        private$theta$mean <- private$theta$mean[oRow, oCol]
+        private$theta$mean <- private$theta$mean[oRow, oCol, drop = FALSE]
         private$Z[[1]] <- private$Z[[1]][, oRow, drop = FALSE]
         private$Z[[2]] <- private$Z[[2]][, oCol, drop = FALSE]
       },

diff --git a/R/R6Class-SimpleSBM_fit.R b/R/R6Class-SimpleSBM_fit.R
@@ -134,7 +134,7 @@ SimpleSBM_fit <-
       reorder = function(){
         o <- order(private$theta$mean %*% private$pi, decreasing = TRUE)
         private$pi <- private$pi[o]
-        private$theta$mean <- private$theta$mean[o,o]
+        private$theta$mean <- private$theta$mean[o, o, drop = FALSE]
         private$Z <- private$Z[, o, drop = FALSE]
       },
       #--------------------------------------------