Merge branch 'ori-2907-custom-dataloader-registry' of https://github.…

…com/scverse/scvi-tools into ori-2907-custom-dataloader-registry
scverse · Jan 9, 2025 · f2c63bd · f2c63bd
2 parents 874935b + 61f2e27
commit f2c63bd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/scvi/model/_scvi.py b/src/scvi/model/_scvi.py
@@ -175,7 +175,8 @@ def __init__(
 
             n_batch = self.summary_stats.n_batch
             use_size_factor_key = self.registry_["setup_args"][
-                f"{REGISTRY_KEYS.SIZE_FACTOR_KEY}_key"]
+                f"{REGISTRY_KEYS.SIZE_FACTOR_KEY}_key"
+            ]
             library_log_means, library_log_vars = None, None
             if (
                 not use_size_factor_key

diff --git a/tests/dataloaders/test_custom_dataloader.py b/tests/dataloaders/test_custom_dataloader.py
@@ -214,16 +214,15 @@ def test_lamindb_dataloader_scvi(save_path: str):
 
     model.save("lamin_model", save_anndata=False, overwrite=True)
     model_query = model.load_query_data(
-        adata=False,
-        reference_model="lamin_model",
-        registry=datamodule.registry)
+        adata=False, reference_model="lamin_model", registry=datamodule.registry
+    )
     model_query.train(max_epochs=1, datamodule=datamodule)
     _ = model_query.get_elbo(dataloader=dataloader)
     _ = model_query.get_marginal_ll(dataloader=dataloader)
     _ = model_query.get_reconstruction_error(dataloader=dataloader)
     _ = model_query.get_latent_representation(dataloader=dataloader)
 
-    adata = collection.load(join='inner')
+    adata = collection.load(join="inner")
     model_query_adata = model.load_query_data(adata=adata, reference_model="lamin_model")
     adata = collection.load(join="inner")
     model_query_adata = model.load_query_data(adata)