From b2bc05e3c71cad807db5048234079546820db440 Mon Sep 17 00:00:00 2001
From: Fontrodona Nicolas <nicolas.fontrodona@ens-lyon.fr>
Date: Mon, 21 Feb 2022 09:36:02 +0100
Subject: [PATCH] src/bed_handler/filter_bed.py: change in filter_bed types
---
src/bed_handler/filter_bed.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/bed_handler/filter_bed.py b/src/bed_handler/filter_bed.py
index b88f833..b982e1a 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 \
--
GitLab