fix: exp list ref heads handling

Abhisekh97 · Nov 16, 2022 · d52e1aa · d52e1aa
1 parent 65f61ac
commit d52e1aa
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/dvc/commands/experiments/ls.py b/dvc/commands/experiments/ls.py
@@ -18,15 +18,13 @@ def run(self):
         )
         tags = self.repo.scm.describe(exps)
         remained = {baseline for baseline, tag in tags.items() if tag is None}
-        base = "refs/heads/"
+        base = "refs/heads"
         ref_heads = self.repo.scm.describe(remained, base=base)
 
         for baseline in exps:
-            name = (
-                tags[baseline]
-                or ref_heads[baseline][len(base) :]
-                or baseline[:7]
-            )
+            name = baseline[:7]
+            if tags[baseline] or ref_heads[baseline]:
+                name = tags[baseline] or ref_heads[baseline][len(base) + 1 :]
             if not name_only:
                 print(f"{name}:")
             for exp_name in exps[baseline]: