run only one DefenseFinder folder

mdmparis · Jan 29, 2025 · 7be6509 · 7be6509
1 parent a70df54
commit 7be6509
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/defense_finder/__init__.py b/defense_finder/__init__.py
@@ -8,10 +8,9 @@ def run(protein_file_name, dbtype, workers, coverage, adf, adf_only, tmp_dir, mo
     scripts=[]
 
     if adf_only == False:
-        gen_args = ['--db-type', dbtype, '--sequence-db', protein_file_name, '--models', 'defense-finder-models/DefenseFinder_{i}', 'all',
-                    '--out-dir', os.path.join(tmp_dir, 'DF_{i}'), '--w', str(workers),
-                    '--coverage-profile', str(coverage), '--exchangeable-weight', '1']
-        scripts = [[f.format(i=i) for f in gen_args] for i in range(1, 6)]
+        scripts.append(['--db-type', dbtype, '--sequence-db',protein_file_name, '--models', 'defense-finder-models/DefenseFinder', 'all',
+                        '--out-dir', os.path.join(tmp_dir, 'DefenseFinder'), '--w', str(workers),
+                        '--coverage-profile', str(coverage), '--exchangeable-weight', '1'])
 
         scripts.append(['--db-type', dbtype, '--sequence-db',protein_file_name, '--models', 'defense-finder-models/RM', 'all',
                         '--out-dir', os.path.join(tmp_dir, 'RM'), '--w', str(workers),