From 19a2ecaa190eccd7ef0f0a791adeb65651780e42 Mon Sep 17 00:00:00 2001 From: Pablo Garay Date: Wed, 5 Feb 2025 10:59:50 -0800 Subject: [PATCH] Add missing import for DGXCloudExecutor Signed-off-by: Pablo Garay --- src/nemo_run/core/execution/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/nemo_run/core/execution/__init__.py b/src/nemo_run/core/execution/__init__.py index 7f2584b..6dc3507 100644 --- 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"]