feat(slurm): add reservation support to job configuration (#147)

- Updated SlurmDeployment to include a new "reservation" field in the job configuration, allowing users to specify job reservations.
- Modified job.sh.j2 template to conditionally include the reservation directive in the SLURM job script if a reservation is provided.

This enhancement improves the flexibility of job submissions in SLURM environments.

Author: mkuznet1

src/madengine/deployment/slurm.py

@@ -676,6 +676,7 @@ def debug(self, msg):
             "network_interface": self.slurm_config.get("network_interface"),
             "exclusive": self.slurm_config.get("exclusive", True),
             "exclude": self.slurm_config.get("exclude"),
+            "reservation": self.slurm_config.get("reservation"),
             "constraint": self.slurm_config.get("constraint"),
             "nodelist": self._normalize_nodelist(self.slurm_config.get("nodelist")),
             "qos": self.slurm_config.get("qos"),

src/madengine/deployment/templates/slurm/job.sh.j2

@@ -8,6 +8,9 @@
 #SBATCH --ntasks-per-node=1
 #SBATCH --gpus-per-node={{ gpus_per_node }}
 #SBATCH --time={{ time_limit }}
+{% if reservation %}
+#SBATCH --reservation={{ reservation }}
+{% endif %}
 {% if nodelist %}
 #SBATCH --nodelist={{ nodelist }}
 {% endif %}