Allow .fna host extension as well

sunbeam-labs · Jun 6, 2024 · 1ddee2b · 1ddee2b
1 parent 179092b
commit 1ddee2b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/sbx_mapping.smk b/sbx_mapping.smk
@@ -28,6 +28,9 @@ else:
     if not GenomeFiles:
         GenomeFiles = [f for f in Cfg["sbx_mapping"]["genomes_fp"].glob("*.fa")]
         HOST_FILE_EXT = ".fa"
+    if not GenomeFiles:
+        GenomeFiles = [f for f in Cfg["sbx_mapping"]["genomes_fp"].glob("*.fna")]
+        HOST_FILE_EXT = ".fna"
     GenomeSegments = {
         PurePath(g.name).stem: read_seq_ids(Cfg["sbx_mapping"]["genomes_fp"] / g)
         for g in GenomeFiles