Update denoise.py

cantinilab · Dec 10, 2024 · 3729bfb · 3729bfb
1 parent 40d6c1d
commit 3729bfb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scprint/tasks/denoise.py b/scprint/tasks/denoise.py
@@ -136,7 +136,7 @@ def __call__(self, model: torch.nn.Module, adata: AnnData):
         self.genes = (
             model.pos.cpu().numpy()
             if self.how != "most var"
-            else list(set(model.genes) & set(genelist))
+            else list(set(model.genes) & set(self.genelist))
         )
         tokeep = None
         metrics = None