From d2952dd803cdaa61b48c510ac512a44b9d2501e3 Mon Sep 17 00:00:00 2001
From: jkobject <jkobject@gmail.com>
Date: Wed, 4 Sep 2024 19:43:12 +0200
Subject: [PATCH] trying precising the attenion type

---
 tests/test_base.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/test_base.py b/tests/test_base.py
index 23c0efc..0e185a8 100644
--- 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,