Skip to content

Commit

Permalink
Iterable to list and import from future
Browse files Browse the repository at this point in the history
  • Loading branch information
eroell committed Dec 6, 2023
1 parent adaa53b commit 02148f5
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions ehrapy/tools/feature_ranking/_rank_features_groups.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
from __future__ import annotations

from collections.abc import Iterable
from typing import Literal, Optional, Union

Expand Down Expand Up @@ -257,7 +259,7 @@ def rank_features_groups(
correction_method: _method_options._correction_method = "benjamini-hochberg",
tie_correct: bool = False,
layer: Optional[str] = None,
rank_obs_columns: Optional[Union[list[str], str]] = None,
rank_obs_columns: Optional[Union[Iterable[str], str]] = None,
**kwds,
) -> None: # pragma: no cover
"""Rank features for characterizing groups.
Expand Down Expand Up @@ -355,7 +357,7 @@ def rank_features_groups(
# rank_obs_columns.remove(groupby)

# move obs columns to X
adata_with_moved_columns = move_to_x(adata, rank_obs_columns)
adata_with_moved_columns = move_to_x(adata, list(rank_obs_columns))

# remove columns previously in X
columns_to_select = adata_with_moved_columns.var_names.difference(adata.var_names)
Expand Down

0 comments on commit 02148f5

Please sign in to comment.