Live metrics config (#9433)

* exp: live metrics config token and offline

* refactor live metrics config

* use get_studio_config

* revert to using env vars for studio config

* update dvc-studio-client to support live metrics config changes

* upgrade dvc-studio-client

* don't change environ when posting live metrics events
This commit is contained in:
Dave Berenbaum
2023-05-17 09:21:01 +00:00
committed by GitHub
parent 99daf5d113
commit 7f6890ff25
9 changed files with 165 additions and 58 deletions
+7 -8
View File
@@ -17,8 +17,7 @@ from typing import (
Union,
)
from dvc_studio_client.env import STUDIO_REPO_URL, STUDIO_TOKEN
from dvc_studio_client.post_live_metrics import get_studio_token_and_repo_url
from dvc_studio_client.post_live_metrics import get_studio_config
from funcy import retry
from dvc.dependency import ParamsDependency
@@ -39,6 +38,7 @@ from dvc.repo.experiments.utils import (
)
from dvc.ui import ui
from dvc.utils.objects import cached_property
from dvc.utils.studio import config_to_env
from .utils import get_remote_executor_refs
@@ -421,12 +421,11 @@ class BaseStashQueue(ABC):
if resume_rev:
run_env[DVCLIVE_RESUME] = "1"
studio_token, studio_repo_url = get_studio_token_and_repo_url()
if studio_token is not None:
run_env[STUDIO_TOKEN] = studio_token
if studio_repo_url is not None:
run_env[STUDIO_REPO_URL] = studio_repo_url
# save studio config to read later by dvc and dvclive
studio_config = get_studio_config(
dvc_studio_config=self.repo.config.get("studio")
)
run_env = {**config_to_env(studio_config), **run_env}
self._pack_args(*args, run_env=run_env, **kwargs)
# save experiment as a stash commit
msg = self._stash_msg(