fix compat function

scverse · Feb 5, 2022 · 5d0db8c · 5d0db8c
1 parent 1f9ccc5
commit 5d0db8c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scvi/data/anndata/_compat.py b/scvi/data/anndata/_compat.py
@@ -165,7 +165,8 @@ def manager_from_setup_dict(
                 field = ProteinObsmField(
                     REGISTRY_KEYS.PROTEIN_EXP_KEY,
                     attr_key,
-                    "_scvi_batch",
+                    use_batch_mask=True,
+                    batch_key="_scvi_batch",
                     colnames_uns_key="_protein_names",
                 )
                 setup_kwargs["protein_expression_obsm_key"] = attr_key