fixes bug when obs index is set

YosefLab · Apr 15, 2024 · e272f29 · e272f29
1 parent ec84c24
commit e272f29
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/mrvi/_model.py b/src/mrvi/_model.py
@@ -397,7 +397,7 @@ def per_sample_inference_fn(pair):
                     mean_zs_,
                     dims=["cell_name", "sample", "latent_dim"],
                     coords={
-                        "cell_name": self.adata.obs_names[indices],
+                        "cell_name": self.adata.obs_names[indices].values,
                         "sample": self.sample_order,
                     },
                     name="sample_representations",
@@ -416,7 +416,7 @@ def per_sample_inference_fn(pair):
                     sampled_zs_,
                     dims=["cell_name", "mc_sample", "sample", "latent_dim"],
                     coords={
-                        "cell_name": self.adata.obs_names[indices],
+                        "cell_name": self.adata.obs_names[indices].values,
                         "sample": self.sample_order,
                     },
                     name="sample_representations",
@@ -562,7 +562,7 @@ def _compute_distance(rep):
                 dists,
                 dims=["cell_name", "sample_x", "sample_y"],
                 coords={
-                    "cell_name": self.adata.obs_names[indices],
+                    "cell_name": self.adata.obs_names[indices].values,
                     "sample_x": self.sample_order,
                     "sample_y": self.sample_order,
                 },
@@ -575,7 +575,7 @@ def _compute_distance(rep):
                 dists,
                 dims=["cell_name", "mc_sample", "sample_x", "sample_y"],
                 coords={
-                    "cell_name": self.adata.obs_names[indices],
+                    "cell_name": self.adata.obs_names[indices].values,
                     "mc_sample": np.arange(reps.shape[1]),
                     "sample_x": self.sample_order,
                     "sample_y": self.sample_order,

diff --git a/tests/test_model.py b/tests/test_model.py
@@ -7,6 +7,7 @@
 @pytest.fixture
 def adata():
     adata = synthetic_iid()
+    adata.obs.index.name = "cell_id"
     adata.obs["sample"] = np.random.choice(15, size=adata.shape[0])
     adata.obs["sample_str"] = [chr(i + ord("a")) for i in adata.obs["sample"]]
     meta1 = np.random.randint(0, 2, size=15)