Merge pull request #123 from pyiron/env_fix

Always include the current path in the PYTHONPATH if it is not already

Author: jan-janssen

pympipool/parallel_executors/mpiexec.py

@@ -1,3 +1,4 @@
+from os.path import abspath
 import pickle
 import sys
 
@@ -33,6 +34,11 @@ def main():
         socket = None
 
     memory = None
+
+    cwd = abspath(".")
+    if cwd not in sys.path:
+        sys.path.insert(1, cwd)
+
     while True:
         # Read from socket
         if mpi_rank_zero:

pympipool/parallel_executors/mpipool.py

@@ -1,3 +1,4 @@
+from os.path import abspath
 import pickle
 import sys
 
@@ -27,6 +28,11 @@ def main():
 
     future_dict = {}
     argument_dict = parse_arguments(argument_lst=sys.argv)
+
+    cwd = abspath(".")
+    if cwd not in sys.path:
+        sys.path.insert(1, cwd)
+
     with MPIPoolExecutor(int(argument_dict["total_cores"])) as executor:
         if executor is not None:
             context, socket = connect_to_socket_interface(