Skip to content

Commit c4c83ab

Browse files
authored
Add image pull secrets param for lepton (#330)
* add image pull secrets for lepton Signed-off-by: Pablo Garay <[email protected]> * update format Signed-off-by: Pablo Garay <[email protected]> --------- Signed-off-by: Pablo Garay <[email protected]>
1 parent 19f7cab commit c4c83ab

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

nemo_run/core/execution/lepton.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,9 @@ class LeptonExecutor(Executor):
5353
node_group: str = ""
5454
mounts: list[dict[str, Any]] = field(default_factory=list)
5555
lepton_job_dir: str = field(init=False, default="")
56+
image_pull_secrets: list[str] = field(
57+
default_factory=list
58+
) # Image pull secrets for container registry authentication
5659
custom_spec: dict[str, Any] = field(default_factory=dict)
5760
pre_launch_commands: list[str] = field(default_factory=list) # Custom commands before launch
5861

@@ -249,7 +252,7 @@ def create_lepton_job(self, name: str):
249252
max_job_failure_retry=None,
250253
envs=envs,
251254
mounts=[Mount(**mount) for mount in self.mounts],
252-
image_pull_secrets=[],
255+
image_pull_secrets=self.image_pull_secrets,
253256
ttl_seconds_after_finished=None,
254257
intra_job_communication=True,
255258
privileged=False,

0 commit comments

Comments
 (0)