rename to queue_size, adopt help & arg

mpc-bioinformatics · Jun 19, 2023 · 371e87e · 371e87e
1 parent b4af8f2
commit 371e87e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/protgraph/cli.py b/protgraph/cli.py
@@ -230,9 +230,8 @@ def _parse_mod(input: str):
         "so that each mass can be converted into integers."
     )
     group.add_argument(
-        "--multithreading_queue_size", "-mtqs", type=int, default=300000,
-        help="Set the size of the queues for the multithreading. Sizes above 30000 have previously caused problems "
-        "on MacOS. Default is set to 300000."
+        "--queue_size", "-qs", type=int, default=30000,
+        help="Set the size of the queues, default is 30000"
     )
 
 

diff --git a/protgraph/protgraph.py b/protgraph/protgraph.py
@@ -32,12 +32,12 @@ def prot_graph(**kwargs):
     if "files" not in kwargs or kwargs["files"] is None:
         raise TypeError("missing argument 'files'")
 
-    multithreading_queue_size = prot_graph_args["multithreading_queue_size"]
+    queue_size = prot_graph_args["queue_size"]
 
     # Set up queues
-    entry_queue = ctx.Queue(multithreading_queue_size)
-    statistics_queue = ctx.Queue(multithreading_queue_size)
-    common_out_file_queue = ctx.Queue(multithreading_queue_size)
+    entry_queue = ctx.Queue(queue_size)
+    statistics_queue = ctx.Queue(queue_size)
+    common_out_file_queue = ctx.Queue(queue_size)
 
     # Get the number of processes.
     number_of_procs = \