fixed broken backwards compat in restrict_to key_added name

Khalid-Usman · May 2, 2019 · d6ca1b8 · d6ca1b8
1 parent 0802a4d
commit d6ca1b8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/scanpy/tools/_leiden.py b/scanpy/tools/_leiden.py
@@ -126,6 +126,8 @@ def leiden(
     # store output into adata.obs
     groups = np.array(part.membership)
     if restrict_to is not None:
+        if key_added == 'louvain':
+            key_added += '_R'
         groups = rename_groups(
             adata,
             key_added,

diff --git a/scanpy/tools/_louvain.py b/scanpy/tools/_louvain.py
@@ -148,6 +148,8 @@ def louvain(
     else:
         raise ValueError('`flavor` needs to be "vtraag" or "igraph" or "taynaud".')
     if restrict_to is not None:
+        if key_added == 'louvain':
+            key_added += '_R'
         groups = rename_groups(
             adata,
             key_added,