Plotting fix

RoheLab · Jul 31, 2024 · 21ea711 · 21ea711
1 parent 8fde63c
commit 21ea711
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/R/plotting.R b/R/plotting.R
@@ -46,6 +46,12 @@ expectation.directed_factor_model <- function(model, ...) {
 #'
 plot_expectation <- function(model) {
   EA <- as.matrix(expectation(model))
+  if (is.null(rownames(EA))) {
+    rownames(EA) <- 1:nrow(EA)
+  }
+  if (is.null(colnames(EA))) {
+    colnames(EA) <- 1:ncol(EA)
+  }
   plot_dense_matrix(EA)
 }