From ebaf8598ae7b3948f46ada1cbabfcee0264c2042 Mon Sep 17 00:00:00 2001 From: "Art.OCathain" Date: Mon, 13 Oct 2025 20:11:53 +0000 Subject: [PATCH] Use uppercase container module name for env var --- vec_inf/client/_slurm_templates.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vec_inf/client/_slurm_templates.py b/vec_inf/client/_slurm_templates.py index 209534d8..751619d5 100644 --- a/vec_inf/client/_slurm_templates.py +++ b/vec_inf/client/_slurm_templates.py @@ -97,7 +97,7 @@ class SlurmScriptTemplate(TypedDict): ], "imports": "source {src_dir}/find_port.sh", "env_vars": [ - f"export {CONTAINER_MODULE_NAME}_BINDPATH=${CONTAINER_MODULE_NAME}_BINDPATH,$(echo /dev/infiniband* | sed -e 's/ /,/g')" + f"export {CONTAINER_MODULE_NAME_UPPER}_BINDPATH=${CONTAINER_MODULE_NAME_UPPER}_BINDPATH,$(echo /dev/infiniband* | sed -e 's/ /,/g')" ], "container_command": f"{CONTAINER_MODULE_NAME} exec --nv {{env_str}} --bind {{model_weights_path}}{{additional_binds}} --containall {IMAGE_PATH} \\", "activate_venv": "source {venv}/bin/activate", @@ -221,7 +221,7 @@ class BatchModelLaunchScriptTemplate(TypedDict): "shebang": "#!/bin/bash\n", "container_setup": f"{CONTAINER_LOAD_CMD}\n", "env_vars": [ - f"export {CONTAINER_MODULE_NAME}_BINDPATH=${CONTAINER_MODULE_NAME}_BINDPATH,$(echo /dev/infiniband* | sed -e 's/ /,/g')" + f"export {CONTAINER_MODULE_NAME_UPPER}_BINDPATH=${CONTAINER_MODULE_NAME_UPPER}_BINDPATH,$(echo /dev/infiniband* | sed -e 's/ /,/g')" ], "server_address_setup": [ "source {src_dir}/find_port.sh",