gto register fix for monorepo (#350)

iterative · Apr 6, 2023 · 092db44 · 092db44
1 parent 737f867
commit 092db44
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/gto/tag.py b/gto/tag.py
@@ -24,6 +24,7 @@
     TAG,
     VERSION,
     Action,
+    name_to_tag,
     tag_re,
     tag_to_name,
 )
@@ -60,6 +61,8 @@ def name_tag(
     if action not in TagTemplates:
         raise UnknownAction(action=action)
 
+    artifact = name_to_tag(artifact)
+
     tag = TagTemplates[action].format(artifact=artifact, version=version, stage=stage)
     if simple:
         return tag

diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -355,6 +355,13 @@ def test_register(repo_with_commit: Tuple[git.Repo, Callable]):
         "❌ Supplied version '1.2.3' cannot be parsed\n",
     )
 
+    _check_successful_cmd(
+        "register",
+        ["-r", repo.working_dir, "classification/dvclive:models/nn"],
+        "classification/dvclive=models/nn@v0.0.1",
+        search_func=_check_output_contains,
+    )
+
 
 def test_assign(repo_with_commit: Tuple[git.Repo, Callable]):
     repo, write_file = repo_with_commit