Fix python paths

SMByC · Aug 23, 2023 · 18f876c · 18f876c
1 parent 32130ec
commit 18f876c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/stack_composed/stack-composed-distributed.py → stack_composed/stack_composed_distributed.py b/stack_composed/stack-composed-distributed.py → stack_composed/stack_composed_distributed.py
@@ -14,13 +14,18 @@
 # systemctl status firewalld
 # systemctl stop firewalld
 
+import os
 from dask.distributed import Client, SSHCluster
 
+# add project dir to pythonpath
+project_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
+if project_dir not in os.sys.path:
+    os.sys.path.append(project_dir)
+
 # get the password from environment variable
-import os
 password = os.environ.get('PASSWORD')
 
-from stack_composed import stack_composed
+from stack_composed import stack_composed_main
 
 # Create a list of addresses for your servers
 addresses = ['192.168.106.12', '192.168.106.13']
@@ -39,7 +44,7 @@
 client = Client(cluster)
 
 # Perform computations using the Dask client
-stack_composed.cli()
+stack_composed_main.cli()
 
 # Close the client and cluster when done
 client.close()

diff --git a/stack_composed/stack_composed.py → stack_composed/stack_composed_main.py b/stack_composed/stack_composed.py → stack_composed/stack_composed_main.py
@@ -27,6 +27,11 @@
 from dask.diagnostics import ProgressBar
 warnings.filterwarnings('ignore')
 
+# add project dir to pythonpath
+project_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
+if project_dir not in os.sys.path:
+    os.sys.path.append(project_dir)
+
 from stack_composed import header
 from stack_composed.image import Image
 from stack_composed.stats import statistic