diff --git a/src/bed_handler/filter_bed.py b/src/bed_handler/filter_bed.py index b88f833831ad1f828c4f3274a5ccd6b501761b26..b982e1ab5791f3568b634f62b130659453bc56d6 100644 --- a/src/bed_handler/filter_bed.py +++ b/src/bed_handler/filter_bed.py @@ -8,7 +8,7 @@ a file, that can be find in a specific column in the bed file """ from pathlib import Path -from typing import List +from typing import List, Union from .config import TestConfig, OutputBed import pandas as pd import lazyparser as lp @@ -30,7 +30,8 @@ def select_ft_of_interest(gene_file: Path) -> List[int]: gene_id for gene_id in gene_list] -def filter_bed(bed_file: Path, gene_list: List[int], col_name: str, keep: bool +def filter_bed(bed_file: Path, gene_list: List[Union[int, float]], + col_name: str, keep: bool ) -> pd.DataFrame: """ load a bed containing FasterDB gene and only recover the gene of \