Merge pull request #636 from Sichao25/debug_l

Support `Index` input for parameter `genes` in  `kinetic_heatmap()`

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:
@@ -180,7 +182,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",

dynamo/prediction/utils.py

@@ -500,7 +500,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: