Renaming

Author: w8jcik

src/moldrug/runner.py

@@ -6,7 +6,7 @@
 
 try:
     from dask.distributed import Client
-    from dask_jobqueue import JobQueueCluster
+    from dask_jobqueue.core import JobQueueCluster
 
     dask_available = True
 except ImportError:
@@ -17,8 +17,8 @@ def serial_execution(func: Callable, entries: List):
     return [func(args) for args in tqdm.tqdm(entries, total=len(entries))]
 
 
-def parallel_execution_processes(func: Callable, entries: List, njobs: int):
-    pool = mp.Pool(njobs)
+def parallel_execution_processes(func: Callable, entries: List, process_count: int):
+    pool = mp.Pool(process_count)
     results = [entry for entry in tqdm.tqdm(pool.imap(func, entries), total=len(entries))]
     pool.close()
 
@@ -75,7 +75,7 @@ def run(self, func: Callable, entries: List):
         if self.mode == RunnerMode.SERIAL:
             return serial_execution(func, entries)
         elif self.mode == RunnerMode.PROCESSES:
-            return parallel_execution_processes(func, entries, njobs=self.process_count)
+            return parallel_execution_processes(func, entries, process_count=self.process_count)
         elif self.mode == RunnerMode.DASK_LOCAL:
             return parallel_execution_dask_local(func, entries, process_count=self.process_count, process_threads_count=self.thread_count)
         if self.mode == RunnerMode.DASK_JOB_QUEUE_CLUSTER: