fixes #242

theislab · Nov 30, 2020 · 95d90de · 95d90de
1 parent f7498c1
commit 95d90de
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/scvelo/plotting/paga.py b/scvelo/plotting/paga.py
@@ -319,9 +319,8 @@ def paga(
                 categories = list(adata.obs[paga_groups].cat.categories)
                 pos = np.zeros((len(categories), 2))
                 for ilabel, label in enumerate(categories):
-                    X_emb = adata.obsm[f"X_{basis}"][
-                        adata.obs[paga_groups] == label, :2
-                    ]
+                    X_emb = adata.obsm[f"X_{basis}"][adata.obs[paga_groups] == label]
+                    X_emb = np.array(X_emb[:, get_components(components, basis)])
                     x_pos, y_pos = np.median(X_emb, axis=0)
                     pos[ilabel] = [x_pos, y_pos]
             else: