Also read .fna files

ssi-dk · Dec 18, 2023 · 081e88f · 081e88f
1 parent acce430
commit 081e88f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bifrost_chewbbaca/run_standalone.py b/bifrost_chewbbaca/run_standalone.py
@@ -60,14 +60,14 @@ def run_pipeline(self):
 
         # Check that we do not get too long sample names!
         for child in input_dir.iterdir():
-            if child.is_file() and child.name.endswith('.fasta'):
+            if child.is_file() and (child.name.endswith('.fasta') or child.name.endswith('.fna')):
                 sample_name = child.name[:-6]
                 if len(sample_name) > 30:
                     print(f"File name too long: {child.name}")
                     sys.exit()
 
         for child in input_dir.iterdir():
-            if child.is_file() and child.name.endswith('.fasta'):
+            if child.is_file() and (child.name.endswith('.fasta') or child.name.endswith('.fna')):
                 sample_count += 1
                 if os.path.isdir(self.output_dir):
                     shutil.rmtree(self.output_dir)