NVIDIA-NeMo · ko3n1g · Mar 6, 2025 · Mar 6, 2025 · Mar 6, 2025 · Mar 6, 2025
diff --git a/.github/workflows/config/codeql.yml b/.github/workflows/config/codeql.yml
@@ -1,7 +1,7 @@
 name: "CodeQL config"
 
 paths:
-  - src/
+  - nemo_run/
   - test/
   - examples/
   - .github/
diff --git a/src/nemo_run/__init__.py → nemo_run/__init__.py b/src/nemo_run/__init__.py → nemo_run/__init__.py
diff --git a/src/nemo_run/__main__.py → nemo_run/__main__.py b/src/nemo_run/__main__.py → nemo_run/__main__.py
diff --git a/src/nemo_run/api.py → nemo_run/api.py b/src/nemo_run/api.py → nemo_run/api.py
diff --git a/src/nemo_run/cli/__init__.py → nemo_run/cli/__init__.py b/src/nemo_run/cli/__init__.py → nemo_run/cli/__init__.py
diff --git a/src/nemo_run/cli/api.py → nemo_run/cli/api.py b/src/nemo_run/cli/api.py → nemo_run/cli/api.py
diff --git a/src/nemo_run/cli/cli_parser.py → nemo_run/cli/cli_parser.py b/src/nemo_run/cli/cli_parser.py → nemo_run/cli/cli_parser.py
diff --git a/src/nemo_run/cli/devspace.py → nemo_run/cli/devspace.py b/src/nemo_run/cli/devspace.py → nemo_run/cli/devspace.py
diff --git a/src/nemo_run/cli/experiment.py → nemo_run/cli/experiment.py b/src/nemo_run/cli/experiment.py → nemo_run/cli/experiment.py
diff --git a/src/nemo_run/cli/workspace.py → nemo_run/cli/workspace.py b/src/nemo_run/cli/workspace.py → nemo_run/cli/workspace.py
diff --git a/src/nemo_run/config.py → nemo_run/config.py b/src/nemo_run/config.py → nemo_run/config.py
diff --git a/src/nemo_run/core/__init__.py → nemo_run/core/__init__.py b/src/nemo_run/core/__init__.py → nemo_run/core/__init__.py
diff --git a/src/nemo_run/core/execution/__init__.py → nemo_run/core/execution/__init__.py b/src/nemo_run/core/execution/__init__.py → nemo_run/core/execution/__init__.py
diff --git a/src/nemo_run/core/execution/base.py → nemo_run/core/execution/base.py b/src/nemo_run/core/execution/base.py → nemo_run/core/execution/base.py
diff --git a/src/nemo_run/core/execution/dgxcloud.py → nemo_run/core/execution/dgxcloud.py b/src/nemo_run/core/execution/dgxcloud.py → nemo_run/core/execution/dgxcloud.py
diff --git a/src/nemo_run/core/execution/docker.py → nemo_run/core/execution/docker.py b/src/nemo_run/core/execution/docker.py → nemo_run/core/execution/docker.py
diff --git a/src/nemo_run/core/execution/launcher.py → nemo_run/core/execution/launcher.py b/src/nemo_run/core/execution/launcher.py → nemo_run/core/execution/launcher.py
diff --git a/src/nemo_run/core/execution/local.py → nemo_run/core/execution/local.py b/src/nemo_run/core/execution/local.py → nemo_run/core/execution/local.py
diff --git a/src/nemo_run/core/execution/skypilot.py → nemo_run/core/execution/skypilot.py b/src/nemo_run/core/execution/skypilot.py → nemo_run/core/execution/skypilot.py
diff --git a/src/nemo_run/core/execution/slurm.py → nemo_run/core/execution/slurm.py b/src/nemo_run/core/execution/slurm.py → nemo_run/core/execution/slurm.py
diff --git a/...n/core/execution/templates/ft_launcher.j2 → ...n/core/execution/templates/ft_launcher.j2 b/...n/core/execution/templates/ft_launcher.j2 → ...n/core/execution/templates/ft_launcher.j2
diff --git a/..._run/core/execution/templates/slurm.sh.j2 → ..._run/core/execution/templates/slurm.sh.j2 b/..._run/core/execution/templates/slurm.sh.j2 → ..._run/core/execution/templates/slurm.sh.j2
diff --git a/.../core/execution/templates/slurm_ray.sh.j2 → .../core/execution/templates/slurm_ray.sh.j2 b/.../core/execution/templates/slurm_ray.sh.j2 → .../core/execution/templates/slurm_ray.sh.j2
diff --git a/src/nemo_run/core/execution/utils.py → nemo_run/core/execution/utils.py b/src/nemo_run/core/execution/utils.py → nemo_run/core/execution/utils.py
diff --git a/src/nemo_run/core/frontend/__init__.py → nemo_run/core/frontend/__init__.py b/src/nemo_run/core/frontend/__init__.py → nemo_run/core/frontend/__init__.py
diff --git a/...emo_run/core/frontend/console/__init__.py → nemo_run/core/frontend/console/__init__.py b/...emo_run/core/frontend/console/__init__.py → nemo_run/core/frontend/console/__init__.py
diff --git a/src/nemo_run/core/frontend/console/api.py → nemo_run/core/frontend/console/api.py b/src/nemo_run/core/frontend/console/api.py → nemo_run/core/frontend/console/api.py
diff --git a/src/nemo_run/core/frontend/console/styles.py → nemo_run/core/frontend/console/styles.py b/src/nemo_run/core/frontend/console/styles.py → nemo_run/core/frontend/console/styles.py
diff --git a/src/nemo_run/core/packaging/__init__.py → nemo_run/core/packaging/__init__.py b/src/nemo_run/core/packaging/__init__.py → nemo_run/core/packaging/__init__.py
diff --git a/src/nemo_run/core/packaging/base.py → nemo_run/core/packaging/base.py b/src/nemo_run/core/packaging/base.py → nemo_run/core/packaging/base.py
diff --git a/src/nemo_run/core/packaging/git.py → nemo_run/core/packaging/git.py b/src/nemo_run/core/packaging/git.py → nemo_run/core/packaging/git.py
diff --git a/src/nemo_run/core/packaging/hybrid.py → nemo_run/core/packaging/hybrid.py b/src/nemo_run/core/packaging/hybrid.py → nemo_run/core/packaging/hybrid.py
diff --git a/src/nemo_run/core/packaging/pattern.py → nemo_run/core/packaging/pattern.py b/src/nemo_run/core/packaging/pattern.py → nemo_run/core/packaging/pattern.py
diff --git a/src/nemo_run/core/runners/__init__.py → nemo_run/core/runners/__init__.py b/src/nemo_run/core/runners/__init__.py → nemo_run/core/runners/__init__.py
diff --git a/src/nemo_run/core/runners/fdl_runner.py → nemo_run/core/runners/fdl_runner.py b/src/nemo_run/core/runners/fdl_runner.py → nemo_run/core/runners/fdl_runner.py
diff --git a/src/nemo_run/core/serialization/__init__.py → nemo_run/core/serialization/__init__.py b/src/nemo_run/core/serialization/__init__.py → nemo_run/core/serialization/__init__.py
diff --git a/src/nemo_run/core/serialization/yaml.py → nemo_run/core/serialization/yaml.py b/src/nemo_run/core/serialization/yaml.py → nemo_run/core/serialization/yaml.py
diff --git a/src/nemo_run/core/serialization/zlib_json.py → nemo_run/core/serialization/zlib_json.py b/src/nemo_run/core/serialization/zlib_json.py → nemo_run/core/serialization/zlib_json.py
diff --git a/src/nemo_run/core/tunnel/__init__.py → nemo_run/core/tunnel/__init__.py b/src/nemo_run/core/tunnel/__init__.py → nemo_run/core/tunnel/__init__.py
diff --git a/src/nemo_run/core/tunnel/client.py → nemo_run/core/tunnel/client.py b/src/nemo_run/core/tunnel/client.py → nemo_run/core/tunnel/client.py
diff --git a/src/nemo_run/core/tunnel/rsync.py → nemo_run/core/tunnel/rsync.py b/src/nemo_run/core/tunnel/rsync.py → nemo_run/core/tunnel/rsync.py
diff --git a/src/nemo_run/core/tunnel/server.py → nemo_run/core/tunnel/server.py b/src/nemo_run/core/tunnel/server.py → nemo_run/core/tunnel/server.py
diff --git a/src/nemo_run/devspace/__init__.py → nemo_run/devspace/__init__.py b/src/nemo_run/devspace/__init__.py → nemo_run/devspace/__init__.py
diff --git a/src/nemo_run/devspace/base.py → nemo_run/devspace/base.py b/src/nemo_run/devspace/base.py → nemo_run/devspace/base.py
diff --git a/src/nemo_run/devspace/editor.py → nemo_run/devspace/editor.py b/src/nemo_run/devspace/editor.py → nemo_run/devspace/editor.py
diff --git a/src/nemo_run/exceptions.py → nemo_run/exceptions.py b/src/nemo_run/exceptions.py → nemo_run/exceptions.py
diff --git a/src/nemo_run/help.py → nemo_run/help.py b/src/nemo_run/help.py → nemo_run/help.py
diff --git a/src/nemo_run/lazy.py → nemo_run/lazy.py b/src/nemo_run/lazy.py → nemo_run/lazy.py
diff --git a/src/nemo_run/run/__init__.py → nemo_run/run/__init__.py b/src/nemo_run/run/__init__.py → nemo_run/run/__init__.py
diff --git a/src/nemo_run/run/api.py → nemo_run/run/api.py b/src/nemo_run/run/api.py → nemo_run/run/api.py
diff --git a/src/nemo_run/run/experiment.py → nemo_run/run/experiment.py b/src/nemo_run/run/experiment.py → nemo_run/run/experiment.py
diff --git a/src/nemo_run/run/job.py → nemo_run/run/job.py b/src/nemo_run/run/job.py → nemo_run/run/job.py
diff --git a/src/nemo_run/run/logs.py → nemo_run/run/logs.py b/src/nemo_run/run/logs.py → nemo_run/run/logs.py
diff --git a/src/nemo_run/run/plugin.py → nemo_run/run/plugin.py b/src/nemo_run/run/plugin.py → nemo_run/run/plugin.py
diff --git a/src/nemo_run/run/task.py → nemo_run/run/task.py b/src/nemo_run/run/task.py → nemo_run/run/task.py
diff --git a/src/nemo_run/run/torchx_backend/__init__.py → nemo_run/run/torchx_backend/__init__.py b/src/nemo_run/run/torchx_backend/__init__.py → nemo_run/run/torchx_backend/__init__.py
diff --git a/...run/torchx_backend/components/__init__.py → ...run/torchx_backend/components/__init__.py b/...run/torchx_backend/components/__init__.py → ...run/torchx_backend/components/__init__.py
diff --git a/.../torchx_backend/components/ft_launcher.py → .../torchx_backend/components/ft_launcher.py b/.../torchx_backend/components/ft_launcher.py → .../torchx_backend/components/ft_launcher.py
diff --git a/...run/torchx_backend/components/torchrun.py → ...run/torchx_backend/components/torchrun.py b/...run/torchx_backend/components/torchrun.py → ...run/torchx_backend/components/torchrun.py
diff --git a/src/nemo_run/run/torchx_backend/launcher.py → nemo_run/run/torchx_backend/launcher.py b/src/nemo_run/run/torchx_backend/launcher.py → nemo_run/run/torchx_backend/launcher.py
diff --git a/src/nemo_run/run/torchx_backend/packaging.py → nemo_run/run/torchx_backend/packaging.py b/src/nemo_run/run/torchx_backend/packaging.py → nemo_run/run/torchx_backend/packaging.py
diff --git a/src/nemo_run/run/torchx_backend/runner.py → nemo_run/run/torchx_backend/runner.py b/src/nemo_run/run/torchx_backend/runner.py → nemo_run/run/torchx_backend/runner.py
diff --git a/...run/torchx_backend/schedulers/__init__.py → ...run/torchx_backend/schedulers/__init__.py b/...run/torchx_backend/schedulers/__init__.py → ...run/torchx_backend/schedulers/__init__.py
diff --git a/..._run/run/torchx_backend/schedulers/api.py → ..._run/run/torchx_backend/schedulers/api.py b/..._run/run/torchx_backend/schedulers/api.py → ..._run/run/torchx_backend/schedulers/api.py
diff --git a/...run/torchx_backend/schedulers/dgxcloud.py → ...run/torchx_backend/schedulers/dgxcloud.py b/...run/torchx_backend/schedulers/dgxcloud.py → ...run/torchx_backend/schedulers/dgxcloud.py
diff --git a/...n/run/torchx_backend/schedulers/docker.py → ...n/run/torchx_backend/schedulers/docker.py b/...n/run/torchx_backend/schedulers/docker.py → ...n/run/torchx_backend/schedulers/docker.py
diff --git a/...un/run/torchx_backend/schedulers/local.py → ...un/run/torchx_backend/schedulers/local.py b/...un/run/torchx_backend/schedulers/local.py → ...un/run/torchx_backend/schedulers/local.py
diff --git a/...run/torchx_backend/schedulers/skypilot.py → ...run/torchx_backend/schedulers/skypilot.py b/...run/torchx_backend/schedulers/skypilot.py → ...run/torchx_backend/schedulers/skypilot.py
diff --git a/...un/run/torchx_backend/schedulers/slurm.py → ...un/run/torchx_backend/schedulers/slurm.py b/...un/run/torchx_backend/schedulers/slurm.py → ...un/run/torchx_backend/schedulers/slurm.py
diff --git a/...un/run/torchx_backend/schedulers/utils.py → ...un/run/torchx_backend/schedulers/utils.py b/...un/run/torchx_backend/schedulers/utils.py → ...un/run/torchx_backend/schedulers/utils.py
diff --git a/src/nemo_run/run/utils.py → nemo_run/run/utils.py b/src/nemo_run/run/utils.py → nemo_run/run/utils.py
diff --git a/pyproject.toml b/pyproject.toml
@@ -90,8 +90,8 @@ addopts = '-m "not slow"'
 
 [tool.coverage.run]
 branch = true
-include = ["src/nemo_run/**/*.py"]
-omit = ["src/nemo_run/core/tunnel/callback.py", "src/nemo_run/help.py", "src/nemo_run/**/__init__.py", "src/nemo_run/**/_version.py"]
+include = ["nemo_run/**/*.py"]
+omit = ["nemo_run/core/tunnel/callback.py", "nemo_run/help.py", "nemo_run/**/__init__.py", "nemo_run/**/_version.py"]
 
 [tool.coverage.report]
 # Regexes for lines to exclude from consideration
@@ -122,13 +122,13 @@ directory = "coverage_html_report"
 allow-direct-references = true
 
 [tool.hatch.build.targets.wheel]
-packages = ["src/nemo_run"]
+packages = ["nemo_run"]
 
 [tool.hatch.version]
 source = "vcs"
 
 [tool.hatch.build.hooks.vcs]
-version-file = "src/nemo_run/_version.py"
+version-file = "nemo_run/_version.py"
 
 [tool.ruff]
 line-length = 100
diff --git a/test/core/execution/test_local.py b/test/core/execution/test_local.py
@@ -16,7 +16,7 @@
 import os
 import tempfile
 
-from src.nemo_run.core.execution.local import LocalExecutor
+from nemo_run.core.execution.local import LocalExecutor
 
 
 def test_local_executor_init():

diff --git a/test/core/execution/test_utils.py b/test/core/execution/test_utils.py
@@ -15,7 +15,7 @@
 
 import pytest
 
-from src.nemo_run.core.execution.utils import fill_template
+from nemo_run.core.execution.utils import fill_template
 
 
 def test_fill_template_file_not_found():

diff --git a/test/core/packaging/test_base.py b/test/core/packaging/test_base.py
@@ -14,9 +14,9 @@
 # limitations under the License.
 
 import pytest
-from nemo_run.config import Config
 
-from src.nemo_run.core.packaging.base import Packager
+from nemo_run.config import Config
+from nemo_run.core.packaging.base import Packager
 
 
 @pytest.fixture