(fix): do proper in-place update

scverse · Oct 15, 2024 · 0afc23a · 0afc23a
1 parent 79a20a8
commit 0afc23a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/muon/_core/preproc.py b/muon/_core/preproc.py
@@ -760,7 +760,7 @@ def func(x):
         for m in data.mod.keys():
             obsmap = data.obsmap[m][obs_subset]
             obsidx = obsmap > 0
-            data.mod[m] = data.mod[m][data.mod[m].obs_names[obsmap[obsidx] - 1]]
+            data.mod[m]._obs = data.mod[m].obs.loc[data.mod[m].obs_names[obsmap[obsidx] - 1]]
             maporder = np.argsort(obsmap[obsidx])
             nobsmap = np.empty(maporder.size)
             nobsmap[maporder] = np.arange(1, maporder.size + 1)