Skip to content

Commit 6bc4319

Browse files
authored
Create SkypilotJobsExecutor to allow running managed jobs (#343)
* Create SkypilotJobsExecutor to allow running managed jobs with Skypilot API Signed-off-by: Rahim Dharssi <[email protected]> * Remove unnecessary comments Signed-off-by: Rahim Dharssi <[email protected]> * fix lints Signed-off-by: Rahim Dharssi <[email protected]> * Add comment for suppressing import error Signed-off-by: Rahim Dharssi <[email protected]> * Write unit tests for _save_job_dir and _get_job_dirs Signed-off-by: Rahim Dharssi <[email protected]> * Fix lints Signed-off-by: Rahim Dharssi <[email protected]> --------- Signed-off-by: Rahim Dharssi <[email protected]>
1 parent f5970c5 commit 6bc4319

File tree

8 files changed

+1414
-0
lines changed

8 files changed

+1414
-0
lines changed

nemo_run/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
from nemo_run.core.execution.local import LocalExecutor
3030
from nemo_run.core.execution.skypilot import SkypilotExecutor
3131
from nemo_run.core.execution.slurm import SlurmExecutor
32+
from nemo_run.core.execution.skypilot_jobs import SkypilotJobsExecutor
3233
from nemo_run.core.packaging import GitArchivePackager, HybridPackager, Packager, PatternPackager
3334
from nemo_run.core.tunnel.client import LocalTunnel, SSHTunnel
3435
from nemo_run.devspace.base import DevSpace
@@ -68,6 +69,7 @@
6869
"run",
6970
"Script",
7071
"SkypilotExecutor",
72+
"SkypilotJobsExecutor",
7173
"SlurmExecutor",
7274
"SSHTunnel",
7375
"Torchrun",

0 commit comments

Comments
 (0)