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]] = {