artifacts: get studio info from local config

This commit is contained in:
Dave Berenbaum
2024-05-23 13:45:48 -04:00
committed by Dave Berenbaum
parent 05db79e7cf
commit ef507efa04
2 changed files with 25 additions and 8 deletions
+9
View File
@@ -281,3 +281,12 @@ def test_artifacts_download_subrepo(tmp_dir, scm, sub):
assert Artifacts.get(subdir.fs_path, name, force=True) == result
with subdir.chdir():
assert Artifacts.get(".", name, force=True) == result
def test_artifacts_download_studio(tmp_dir, dvc, mocker):
with dvc.config.edit("global") as conf:
conf["studio"]["token"] = "mytoken"
download_studio = mocker.patch("dvc.repo.artifacts.Artifacts._download_studio")
Artifacts.get("myart.pkl", "myart.pkl")
assert download_studio.call_args.kwargs["dvc_studio_config"]["token"] == "mytoken"