trying precising the attenion type

jkobject · Sep 4, 2024 · d2952dd · d2952dd
1 parent db8ef43
commit d2952dd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_base.py b/tests/test_base.py
@@ -39,7 +39,9 @@ def test_base():
     adata = preprocessor(adata)
     # conf = dict(self.config_init[subcommand])
 
-    model = scPrint.load_from_checkpoint(ckpt_path, precpt_gene_emb=None)
+    model = scPrint.load_from_checkpoint(
+        ckpt_path, precpt_gene_emb=None, attention_type="normal"
+    )
     dn = Denoiser(
         plot_corr_size=10,
         batch_size=2,