Use exclude_calls_file_cols

PMBio · Nov 28, 2023 · becea82 · becea82
1 parent 448eab5
commit becea82
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/deeprvat/preprocessing/preprocess.py b/deeprvat/preprocessing/preprocess.py
@@ -253,12 +253,13 @@ def process_sparse_gt(
         logging.info(f"Processing chromosome {chrom}")
         logging.info("Reading in filtered calls")
 
+        exclude_calls_file_cols = ["chrom", "pos", "ref", "alt", "sample"]
+
         calls_to_exclude = pd.DataFrame(
-            columns=["chrom", "pos", "ref", "alt", "sample"]
+            columns=exclude_calls_file_cols
         )
 
         if exclude_calls is not None:
-            exclude_calls_file_cols = ["chrom", "pos", "ref", "alt", "sample"]
 
             exclude_calls_chrom = Path(exclude_calls).rglob("*.tsv*")
             exclude_calls_chrom_files = []