From 9b834c640bf0a9efeef117ce0dd6820db48d26a7 Mon Sep 17 00:00:00 2001 From: David de la Iglesia Castro Date: Mon, 13 Mar 2023 11:43:35 +0100 Subject: [PATCH] exp: Include `STUDIO_REPO_URL` in run_env. Closes #8951. --- dvc/repo/experiments/queue/base.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/dvc/repo/experiments/queue/base.py b/dvc/repo/experiments/queue/base.py index 82d8061818..72913d59df 100644 --- a/dvc/repo/experiments/queue/base.py +++ b/dvc/repo/experiments/queue/base.py @@ -17,6 +17,8 @@ Union, ) +from dvc_studio_client.env import STUDIO_REPO_URL +from dvc_studio_client.post_live_metrics import get_studio_repo_url from funcy import retry from dvc.dependency import ParamsDependency @@ -415,6 +417,10 @@ def _stash_exp( # noqa: PLR0915, C901 run_env[DVC_EXP_NAME] = name if resume_rev: run_env[DVCLIVE_RESUME] = "1" + studio_repo_url = get_studio_repo_url() + if studio_repo_url is not None: + run_env[STUDIO_REPO_URL] = studio_repo_url + self._pack_args(*args, run_env=run_env, **kwargs) # save experiment as a stash commit