Add missing import for DGXCloudExecutor

Signed-off-by: Pablo Garay <palenq@gmail.com>
NVIDIA · Feb 5, 2025 · 19a2eca · 19a2eca
1 parent 570f577
commit 19a2eca
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/nemo_run/core/execution/__init__.py b/src/nemo_run/core/execution/__init__.py
@@ -16,5 +16,6 @@
 from nemo_run.core.execution.local import LocalExecutor
 from nemo_run.core.execution.skypilot import SkypilotExecutor
 from nemo_run.core.execution.slurm import SlurmExecutor
+from nemo_run.core.execution.dgxcloud import DGXCloudExecutor
 
 __all__ = ["LocalExecutor", "SlurmExecutor", "SkypilotExecutor", "DGXCloudExecutor"]