From 3f1b0daf11a4c73198b894f3d58203578df9cd6c Mon Sep 17 00:00:00 2001
From: Yannick DAYER <yannick.dayer@idiap.ch>
Date: Tue, 26 Mar 2024 11:13:44 +0100
Subject: [PATCH] fix(slurm): add the worker-port conf for our setup

---
 src/bob/pipelines/config/distributed/slurm_cpu_default.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/bob/pipelines/config/distributed/slurm_cpu_default.py b/src/bob/pipelines/config/distributed/slurm_cpu_default.py
index 74105ca..be6aa28 100644
--- a/src/bob/pipelines/config/distributed/slurm_cpu_default.py
+++ b/src/bob/pipelines/config/distributed/slurm_cpu_default.py
@@ -70,8 +70,12 @@ cluster = SLURMCluster(
     protocol="tcp://",
     scheduler_options={
         "protocol": "tcp://",
-        "port": 8786,  # Workers will connect to the scheduler on that port.
+        "port": 8786,  # Workers will connect to the scheduler on that port
     },
+    worker_extra_args=[
+        "--worker-port",
+        "60001:63000",  # Workers will be reachable by the Client on those ports
+    ],
 )
 
 cluster.adapt(
-- 
GitLab