nemo-evaluator-launcher 0.1.64__tar.gz → 0.1.67__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (74) hide show
  1. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/PKG-INFO +1 -1
  2. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/slurm/executor.py +3 -3
  3. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/package_info.py +1 -1
  4. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher.egg-info/PKG-INFO +1 -1
  5. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/LICENSE +0 -0
  6. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/README.md +0 -0
  7. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/pyproject.toml +0 -0
  8. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/setup.cfg +0 -0
  9. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/__init__.py +0 -0
  10. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/api/__init__.py +0 -0
  11. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/api/functional.py +0 -0
  12. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/api/types.py +0 -0
  13. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/api/utils.py +0 -0
  14. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/cli/__init__.py +0 -0
  15. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/cli/export.py +0 -0
  16. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/cli/info.py +0 -0
  17. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/cli/kill.py +0 -0
  18. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/cli/logs.py +0 -0
  19. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/cli/ls_runs.py +0 -0
  20. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/cli/ls_task.py +0 -0
  21. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/cli/ls_tasks.py +0 -0
  22. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/cli/main.py +0 -0
  23. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/cli/run.py +0 -0
  24. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/cli/status.py +0 -0
  25. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/cli/version.py +0 -0
  26. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/common/__init__.py +0 -0
  27. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/common/container_metadata/__init__.py +0 -0
  28. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/common/container_metadata/intermediate_repr.py +0 -0
  29. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/common/container_metadata/loading.py +0 -0
  30. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/common/container_metadata/registries.py +0 -0
  31. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/common/container_metadata/utils.py +0 -0
  32. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/common/execdb.py +0 -0
  33. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/common/helpers.py +0 -0
  34. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/common/logging_utils.py +0 -0
  35. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/common/mapping.py +0 -0
  36. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/common/printing_utils.py +0 -0
  37. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/configs/__init__.py +0 -0
  38. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/configs/default.yaml +0 -0
  39. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/configs/deployment/generic.yaml +0 -0
  40. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/configs/deployment/nim.yaml +0 -0
  41. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/configs/deployment/none.yaml +0 -0
  42. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/configs/deployment/sglang.yaml +0 -0
  43. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/configs/deployment/trtllm.yaml +0 -0
  44. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/configs/deployment/vllm.yaml +0 -0
  45. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/configs/execution/lepton/default.yaml +0 -0
  46. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/configs/execution/local.yaml +0 -0
  47. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/configs/execution/slurm/default.yaml +0 -0
  48. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/__init__.py +0 -0
  49. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/base.py +0 -0
  50. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/lepton/__init__.py +0 -0
  51. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/lepton/deployment_helpers.py +0 -0
  52. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/lepton/executor.py +0 -0
  53. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/lepton/job_helpers.py +0 -0
  54. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/local/__init__.py +0 -0
  55. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/local/executor.py +0 -0
  56. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/local/run.template.sh +0 -0
  57. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/registry.py +0 -0
  58. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/slurm/__init__.py +0 -0
  59. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/slurm/proxy.cfg.template +0 -0
  60. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/exporters/__init__.py +0 -0
  61. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/exporters/base.py +0 -0
  62. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/exporters/gsheets.py +0 -0
  63. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/exporters/local.py +0 -0
  64. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/exporters/mlflow.py +0 -0
  65. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/exporters/registry.py +0 -0
  66. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/exporters/utils.py +0 -0
  67. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/exporters/wandb.py +0 -0
  68. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/resources/all_tasks_irs.yaml +0 -0
  69. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/resources/mapping.toml +0 -0
  70. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher.egg-info/SOURCES.txt +0 -0
  71. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher.egg-info/dependency_links.txt +0 -0
  72. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher.egg-info/entry_points.txt +0 -0
  73. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher.egg-info/requires.txt +0 -0
  74. {nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher.egg-info/top_level.txt +0 -0
{nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nemo-evaluator-launcher
3
- Version: 0.1.64
3
+ Version: 0.1.67
4
4
  Summary: Launcher for the evaluations provided by NeMo Evaluator containers with different runtime backends
5
5
  Author: NVIDIA
6
6
  Author-email: nemo-toolkit@nvidia.com
{nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/executors/slurm/executor.py RENAMED
@@ -710,7 +710,7 @@ def _create_slurm_sbatch_script(
710
710
 
711
711
  s += "# evaluation client\n"
712
712
  s += "srun --mpi pmix --overlap "
713
- s += "--nodes 1 --ntasks 1 " # Client always runs on single node
713
+ s += "--nodelist ${nodes_array[0]} --nodes 1 --ntasks 1 "
714
714
  s += "--container-image {} ".format(eval_image)
715
715
  evaluation_env_var_names = list(
716
716
  cfg.execution.get("env_vars", {}).get("evaluation", {})
@@ -835,7 +835,7 @@ def _generate_auto_export_section(
835
835
 
836
836
  s += " # export\n"
837
837
  s += " srun --mpi pmix --overlap "
838
- s += "--nodes 1 --ntasks 1 " # Client always runs on single node
838
+ s += "--nodelist ${nodes_array[0]} --nodes 1 --ntasks 1 "
839
839
  s += "--container-image {} ".format(export_image)
840
840
  if export_env:
841
841
  s += "--container-env {} ".format(",".join(export_env))
@@ -1577,7 +1577,7 @@ def _generate_haproxy_srun_command(cfg, remote_task_subdir):
1577
1577
  s += "done\n"
1578
1578
  s += "\n"
1579
1579
  s += "srun --mpi pmix --overlap "
1580
- s += "--nodes 1 --ntasks 1 "
1580
+ s += "--nodelist ${nodes_array[0]} --nodes 1 --ntasks 1 "
1581
1581
  s += f"--container-image {cfg.execution.get('proxy', {}).get('image', 'haproxy:latest')} "
1582
1582
  s += f"--container-mounts {remote_task_subdir}/proxy.cfg:/usr/local/etc/haproxy/haproxy.cfg:ro "
1583
1583
  s += f"--output {remote_task_subdir}/logs/proxy-%A.log "
{nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher/package_info.py RENAMED
@@ -16,7 +16,7 @@
16
16
  # Below is the _next_ version that will be published, not the currently published one.
17
17
  MAJOR = 0
18
18
  MINOR = 1
19
- PATCH = 64
19
+ PATCH = 67
20
20
  PRE_RELEASE = ""
21
21
 
22
22
  # Use the following formatting: (major, minor, patch, pre-release)
{nemo_evaluator_launcher-0.1.64 → nemo_evaluator_launcher-0.1.67}/src/nemo_evaluator_launcher.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nemo-evaluator-launcher
3
- Version: 0.1.64
3
+ Version: 0.1.67
4
4
  Summary: Launcher for the evaluations provided by NeMo Evaluator containers with different runtime backends
5
5
  Author: NVIDIA
6
6
  Author-email: nemo-toolkit@nvidia.com