error

jkobject · Sep 4, 2024 · e1b3eb3 · e1b3eb3
1 parent d2952dd
commit e1b3eb3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_base.py b/tests/test_base.py
@@ -40,7 +40,10 @@ def test_base():
     # conf = dict(self.config_init[subcommand])
 
     model = scPrint.load_from_checkpoint(
-        ckpt_path, precpt_gene_emb=None, attention_type="normal"
+        ckpt_path,
+        precpt_gene_emb=None,
+        # triton gets installed so it must think it has cuda enabled
+        transformer="normal",
     )
     dn = Denoiser(
         plot_corr_size=10,