Fix mu.pl.embedding

scverse · Oct 8, 2021 · e40ca58 · e40ca58
1 parent 85fcfe5
commit e40ca58
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/muon/_core/plot.py b/muon/_core/plot.py
@@ -107,7 +107,7 @@ def embedding(
             for i, k in enumerate(keys):
                 for m in data.mod:
                     if keys_in_mod[m][i] == False and data.mod[m].raw is not None:
-                        keys_in_mod[m][i] = True
+                        keys_in_mod[m][i] = k in data.mod[m].raw.var_names
 
         for m in data.mod:
             if np.sum(keys_in_mod[m]) > 0: