Merge pull request #50 from tdiethe/tdiethe-patch-1

Added .fna, .ffn, .faa, .frn as alternative fasta suffixes
songlab-cal · Apr 28, 2020 · 3d3657b · 3d3657b
2 parents c4f5407 + 9b3fc33
commit 3d3657b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tape/datasets.py b/tape/datasets.py
@@ -24,7 +24,7 @@ def dataset_factory(data_file: Union[str, Path], *args, **kwargs) -> Dataset:
         raise FileNotFoundError(data_file)
     if data_file.suffix == '.lmdb':
         return LMDBDataset(data_file, *args, **kwargs)
-    elif data_file.suffix == '.fasta':
+    elif data_file.suffix in {'.fasta', '.fna', '.ffn', '.faa', '.frn'}:
         return FastaDataset(data_file, *args, **kwargs)
     elif data_file.suffix == '.json':
         return JSONDataset(data_file, *args, **kwargs)