If is a string, put it into a list

theislab · Apr 9, 2024 · ee536cd · ee536cd
1 parent cb5e09a
commit ee536cd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scvelo/core/_anndata.py b/scvelo/core/_anndata.py
@@ -206,7 +206,8 @@ def get_df(
         if keys is None:
             df = data.to_df()
         else:
-            key = keys if isinstance(keys, str) else keys[0]
+            keys = [keys] if isinstance(keys, str) else keys
+            key = keys[0]
 
             if key in data.var_names:
                 df = obs_df(data, keys, layer=layer)