making local submit one job at a time

courtois-neuromod · Jan 19, 2024 · 9e227a0 · 9e227a0
1 parent f275d09
commit 9e227a0
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/plugins/hydra_submitit_launcher/hydra_plugins/hydra_submitit_launcher/submitit_launcher.py b/plugins/hydra_submitit_launcher/hydra_plugins/hydra_submitit_launcher/submitit_launcher.py
@@ -143,9 +143,16 @@ def launch(
                     Singleton.get_state(),
                 )
             )
-
-        jobs = executor.map_array(self, *zip(*job_params))
-        return [j.results()[0] for j in jobs]
+
+        if self._EXECUTOR == "local":
+            results = []
+            for i, job_param in enumerate(job_params):
+                job = executor.map_array(self, *zip(*[job_params[i]]))
+                results.append(job[0].results()[0])
+            return results
+        else: # self._EXECUTOR == "false":
+            jobs = executor.map_array(self, *zip(*job_params))
+            return [j.results()[0] for j in jobs]
 
 
 class LocalLauncher(BaseSubmititLauncher):