From df4b3356e4a92aca536d5f2d97184bbc2d34b65c Mon Sep 17 00:00:00 2001 From: Karin Hrovatin <47607471+Hrovatin@users.noreply.github.com> Date: Fri, 1 Nov 2024 17:57:57 +0100 Subject: [PATCH] bugfix in test for checking cov embeding --- tests/external/sysvi/test_sysvi.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/external/sysvi/test_sysvi.py b/tests/external/sysvi/test_sysvi.py index c9b4e8f48c..c47b22a9d1 100644 --- 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))