Merge pull request #855 from rfbgo/workflow_hostfile

Add default variable to define likely hostfile path
GoogleCloudPlatform · Jan 30, 2025 · 9c78b49 · 9c78b49
2 parents 9afa675 + 13fe551
commit 9c78b49
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/lib/ramble/ramble/workflow_manager.py b/lib/ramble/ramble/workflow_manager.py
@@ -56,6 +56,10 @@ class WorkflowManagerBase(metaclass=WorkflowManagerMeta):
         description="Hostfile command to apply within execution templates for the workflow",
     )
 
+    workflow_manager_variable(
+        "hostfile", default="{experiment_run_dir}/hostfile", description="Default hostfile path"
+    )
+
     def __init__(self, file_path):
         super().__init__()
 

diff --git a/var/ramble/repos/builtin/workflow_managers/slurm/workflow_manager.py b/var/ramble/repos/builtin/workflow_managers/slurm/workflow_manager.py
@@ -271,4 +271,4 @@ def get_partitions(self):
         }
 
     def get_hostfile_cmd(self):
-        return "scontrol show hostnames > {experiment_run_dir}/hostfile"
+        return "scontrol show hostnames > {hostfile}"