From dd05e77aaf019e1e030a4addc3a0a04183559649 Mon Sep 17 00:00:00 2001 From: Philipp Weiler Date: Thu, 22 Feb 2024 08:09:25 +0000 Subject: [PATCH] Update `heatmap` Subset genes specified in `gene_order` to ones for which a model has been fit. --- src/cellrank/pl/_heatmap.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/cellrank/pl/_heatmap.py b/src/cellrank/pl/_heatmap.py index be49a7b84..7359593cb 100644 --- a/src/cellrank/pl/_heatmap.py +++ b/src/cellrank/pl/_heatmap.py @@ -362,6 +362,7 @@ def _(gene_order: Optional[Sequence[str]] = None) -> Tuple[List[plt.Figure], pd. if not cluster_genes: if gene_order is not None: + gene_order = [gene for gene in gene_order if gene in df.index] df = df.loc[gene_order] else: max_sort = np.argsort(np.argmax(df.apply(_min_max_scale, axis=1).values, axis=1))