Skip to content
Draft
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions nemo_run/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@
from nemo_run.core.execution.docker import DockerExecutor
from nemo_run.core.execution.kubeflow import KubeflowExecutor
from nemo_run.core.execution.launcher import FaultTolerance, SlurmRay, SlurmTemplate, Torchrun
from nemo_run.core.execution.xcalibur import XCaliburExecutor
from nemo_run.core.execution.lepton import LeptonExecutor
from nemo_run.core.execution.local import LocalExecutor
from nemo_run.core.execution.skypilot import SkypilotExecutor
Expand Down Expand Up @@ -68,6 +69,7 @@
"Partial",
"Plugin",
"KubeflowExecutor",
"XCaliburExecutor",
"run",
"Script",
"SkypilotExecutor",
Expand Down
4 changes: 3 additions & 1 deletion nemo_run/core/execution/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,11 +14,12 @@
# limitations under the License.

from nemo_run.core.execution.dgxcloud import DGXCloudExecutor
from nemo_run.core.execution.kubeflow import KubeflowExecutor
from nemo_run.core.execution.lepton import LeptonExecutor
from nemo_run.core.execution.local import LocalExecutor
from nemo_run.core.execution.kubeflow import KubeflowExecutor
from nemo_run.core.execution.skypilot import SkypilotExecutor
from nemo_run.core.execution.slurm import SlurmExecutor
from nemo_run.core.execution.xcalibur import XCaliburExecutor

__all__ = [
"LocalExecutor",
Expand All @@ -27,4 +28,5 @@
"DGXCloudExecutor",
"LeptonExecutor",
"KubeflowExecutor",
"XCaliburExecutor",
]
Loading
Loading