From 2da8d6fe5960df1a0947c34d6e46e5ce19f0a0df Mon Sep 17 00:00:00 2001 From: Sergey Serebryakov Date: Fri, 8 Jul 2022 08:55:41 -0700 Subject: [PATCH] MLCube environment variables. This commit introduces a new feature for MLCubes - special environmental variables set by MLCube that are available in user scripts. Two variables are set: - `MLCUBE_NAME` - name of MLCube project. - `MLCUBE_TASK_NAME` - name of current MLCube task. --- runners/mlcube_docker/mlcube_docker/docker_run.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/runners/mlcube_docker/mlcube_docker/docker_run.py b/runners/mlcube_docker/mlcube_docker/docker_run.py index d861a245..ea9ac563 100644 --- a/runners/mlcube_docker/mlcube_docker/docker_run.py +++ b/runners/mlcube_docker/mlcube_docker/docker_run.py @@ -160,7 +160,9 @@ def run(self) -> None: logger.info(f"mounts={mounts}, task_args={task_args}") volumes = Shell.to_cli_args(mounts, sep=':', parent_arg='--volume') - env_args = self.mlcube.runner.env_args + env_args: str = self.mlcube.runner.env_args + f" -e MLCUBE_TASK_NAME={self.task}" + if self.mlcube.get('name', None): + env_args += f" -e MLCUBE_NAME={self.mlcube['name']}" num_gpus: int = self.mlcube.platform.get('accelerator_count', None) or 0 run_args: t.Text = self.mlcube.runner.cpu_args if num_gpus == 0 else self.mlcube.runner.gpu_args try: