Merge pull request #281 from pyiron/executor_import

Use pympipool.Executor rather than pympipool.mpi.PyMPIExecutor
pyiron · Apr 21, 2024 · 9074e11 · 9074e11
2 parents 21bc017 + 9277085
commit 9074e11
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/.ci_support/environment-old.yml b/.ci_support/environment-old.yml
@@ -7,5 +7,5 @@ dependencies:
 - jinja2 =2.11.3
 - paramiko =2.7.1
 - tqdm =4.66.1
-- pympipool =0.7.0
+- pympipool =0.8.0
 - cloudpickle =2.2.1
diff --git a/pysqa/executor/backend.py b/pysqa/executor/backend.py
@@ -2,7 +2,7 @@
 from typing import Optional
 import sys
 
-from pympipool.mpi import PyMPIExecutor
+from pympipool import Executor
 from pysqa.executor.helper import (
     read_from_file,
     deserialize,
@@ -42,16 +42,15 @@ def execute_files_from_list(
 
 def execute_tasks(cores: int, cache_directory: str):
     tasks_in_progress_dict = {}
-    with PyMPIExecutor(
-        max_workers=cores,
+    with Executor(
+        max_cores=cores,
         cores_per_worker=1,
         threads_per_core=1,
         gpus_per_worker=0,
         oversubscribe=False,
         init_function=None,
         cwd=cache_directory,
-        sleep_interval=0.1,
-        enable_slurm_backend=False,
+        backend="mpi",
     ) as exe:
         while True:
             execute_files_from_list(