Also test logit normal distribution

lucas-diedrich · Mar 27, 2024 · 9f6e405 · 9f6e405
1 parent 2c8aa3e
commit 9f6e405
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -78,12 +78,12 @@ def test_pretraining_early_stopping(adata, pretraining_early_stopping, requires_
     assert model.module.z_encoder.encoder.fc_layers[0][0].weight.requires_grad == requires_grad
 
 
-@pytest.fixture(scope="module")
-def basic_train(adata):
+@pytest.fixture(scope="module", params=["normal", "ln"])
+def basic_train(adata, request):
     SCCORAL.setup_anndata(
         adata, categorical_covariates="categorical_covariate", continuous_covariates="continuous_covariate"
     )
-    model = SCCORAL(adata, n_latent=5)
+    model = SCCORAL(adata, n_latent=5, latent_distribution=request.param)
     model.train(max_epochs=20, accelerator="cpu")
 
     return model