fix: processes=len(db_list)

bigict · Mar 22, 2024 · f96199c · f96199c
1 parent b95382d
commit f96199c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/msa_build.py b/msa_build.py
@@ -403,7 +403,7 @@ def run_command_fn(cmd_template, cmd_prefix, db_list, db_idx, **kwargs):
 def run_command_mp(cmd_template, cmd_prefix, db_list, **kwargs):
   command_fn = functools.partial(run_command_fn, cmd_template, cmd_prefix,
                                  db_list, **kwargs)
-  with mp.Pool() as p:
+  with mp.Pool(processes=len(db_list)) as p:
     yield from p.imap(command_fn, range(len(db_list)))