exp: Include STUDIO_REPO_URL in run_env.

Closes #8951.
iterative · Mar 13, 2023 · 9b834c6 · 9b834c6
1 parent 16ecd53
commit 9b834c6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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