get label centers for milo

mjz1 · Jul 2, 2024 · bd8df13 · bd8df13
1 parent ef7cc7e
commit bd8df13
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/NEWS.md b/NEWS.md
@@ -2,6 +2,7 @@
 
 * Replace `add_gc_cor`, `segment_cnv`, and `merge_segments` parallel backend to use `BiocParallel`
 * Added `segment=FALSE` as default
+* `get_label_centers` works with `Milo` objects
 * Minor bugfixes
 
 

diff --git a/R/plotting.R b/R/plotting.R
@@ -644,7 +644,7 @@ my_dens <- function(data, mapping, center_point = 0, ...) {
 get_label_centers <- function(obj, group_var = "clusters", reduced_dim = "UMAP") {
   # To any scatter of a umap can add + geom_label_repel(data = get_label_centers(sce), aes(x = x, y = y, label = clusters))
 
-  if ("SingleCellExperiment" %in% class(obj)) {
+  if (class(obj) %in% c("SingleCellExperiment", "Milo")) {
     x_means <- lapply(split(reducedDim(obj, reduced_dim)[, 1], obj[[group_var]]), mean) %>% unlist()
     y_means <- lapply(split(reducedDim(obj, reduced_dim)[, 2], obj[[group_var]]), mean) %>% unlist()
     centers <- data.frame(x = x_means, y = y_means)