Invert (and fix) the tissue_mask when creating tensor object for gene…

… maps analysis (#46) Co-authored-by: Angad Pal Singh <[email protected]>
ludvb · Apr 13, 2022 · c1f7ead · c1f7ead
1 parent 042e9a9
commit c1f7ead
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xfuse/analyze/gene_maps.py b/xfuse/analyze/gene_maps.py
@@ -156,7 +156,7 @@ def _prepare(x):
 
     def _save_tensor(gene, samples, tissue_mask):
         if tissue_mask is not None:
-            samples[:, tissue_mask] = 0.0
+            samples[:, ~tissue_mask] = 0.0
         torch.save(samples, f"{gene}.pt")
 
     writers: Dict[str, Callable[..., None]] = {