debug kinetic_heatmap genes input type

aristoteleo · Dec 12, 2023 · 9a28f44 · 9a28f44
1 parent ea79042
commit 9a28f44
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/dynamo/plot/time_series.py b/dynamo/plot/time_series.py
@@ -1,6 +1,8 @@
 # include pseudotime and predict cell trajectory
 from typing import Any, Dict, List, Optional, Tuple, Union
 
+import pandas as pd
+
 try:
     from typing import Literal
 except ImportError:
@@ -200,7 +202,7 @@ def kinetic_curves(
 @docstrings.with_indent(4)
 def kinetic_heatmap(
     adata: AnnData,
-    genes: List[str],
+    genes: Union[List[str], pd.Index],
     mode: str = "vector_field",
     basis: Optional[str] = None,
     layer: str = "X",

diff --git a/dynamo/prediction/utils.py b/dynamo/prediction/utils.py
@@ -501,6 +501,8 @@ def arclength_sampling_n(X, num, t=None):
 # fate related
 def fetch_exprs(adata, basis, layer, genes, time, mode, project_back_to_high_dim, traj_ind):
     import pandas as pd
+    if type(genes) != list:
+        genes = list(genes)
 
     prefix = "LAP_" if mode.lower() == "lap" else "fate_"
     if basis is not None: