diff --git a/xfuse/analyze/gene_maps.py b/xfuse/analyze/gene_maps.py index a87d8a4..164fcf3 100644 --- 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]] = {