mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-01 23:08:43 +01:00
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:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user