From b98481223f8d88d03e6518ebb8afac415ba52998 Mon Sep 17 00:00:00 2001 From: ansjindal Date: Fri, 22 Aug 2025 15:32:59 +0200 Subject: [PATCH] Add reservation_config support to LeptonExecutor - Add reservation_config field to LeptonExecutor dataclass - Pass reservation_config from executor to job specification - Enable users to configure reservation settings for Lepton jobs Signed-off-by: ansjindal --- nemo_run/core/execution/lepton.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nemo_run/core/execution/lepton.py b/nemo_run/core/execution/lepton.py index e3aec777..4b22b9d8 100644 --- a/nemo_run/core/execution/lepton.py +++ b/nemo_run/core/execution/lepton.py @@ -55,6 +55,7 @@ class LeptonExecutor(Executor): lepton_job_dir: str = field(init=False, default="") custom_spec: dict[str, Any] = field(default_factory=dict) pre_launch_commands: list[str] = field(default_factory=list) # Custom commands before launch + reservation_config: Optional[dict[str, Any]] = None def stop_job(self, job_id: str): """ @@ -232,7 +233,7 @@ def create_lepton_job(self, name: str): log=None, queue_config=None, stopped=None, - reservation_config=None, + reservation_config=self.reservation_config, ) job = LeptonJob(spec=job_spec, metadata=Metadata(id=name))