bugfix in test for checking cov embeding

scverse · Nov 1, 2024 · df4b335 · df4b335
1 parent 87a367e
commit df4b335
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/external/sysvi/test_sysvi.py b/tests/external/sysvi/test_sysvi.py
@@ -91,7 +91,7 @@ def test_sysvi_model(
     # Model
 
     # Check that model runs through with standard normal prior
-    model = SysVI(adata=adata, prior="standard_normal")
+    model = SysVI(adata=adata, prior="standard_normal", embed_cat=embed_cat)
     model.train(max_epochs=2, batch_size=math.ceil(adata.n_obs / 2.0))
 
     # Check that model runs through with vamp prior
@@ -100,6 +100,7 @@ def test_sysvi_model(
         prior="vamp",
         pseudoinputs_data_indices=pseudoinputs_data_indices,
         n_prior_components=5,
+        embed_cat=embed_cat,
     )
     model.train(max_epochs=2, batch_size=math.ceil(adata.n_obs / 2.0))