From 8c50c3c606d072733402cc1f189f08d2d71820e5 Mon Sep 17 00:00:00 2001
From: Julien Chiquet <julien.chiquet@gmail.com>
Date: Thu, 29 Apr 2021 09:49:50 +0200
Subject: [PATCH] added drop = FALSE ton handle the case when Q = 1 in
 reordering

---
 R/R6Class-BipartiteSBM_fit.R | 2 +-
 R/R6Class-SimpleSBM_fit.R    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/R/R6Class-BipartiteSBM_fit.R b/R/R6Class-BipartiteSBM_fit.R
index e2ab1a23..0e0d424c 100644
--- 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
index 62cbf46c..12697213 100644
--- 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]
       },
       #--------------------------------------------