Update dvc/repo/experiments/executor/base.py

Co-authored-by: Peter Rowlands (변기호) <[email protected]>
iterative · Jul 27, 2021 · 770e5bc · 770e5bc
1 parent 14175c8
commit 770e5bc
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/dvc/repo/experiments/executor/base.py b/dvc/repo/experiments/executor/base.py
@@ -471,10 +471,7 @@ def checkpoint_callback(
             if git_remote:
                 from dvc.repo.experiments.push import push
 
-                branch = dvc.experiments.get_branch_by_rev(
-                    exp_rev, allow_multiple=None
-                )
-                branch_name = ExpRefInfo.from_ref(branch).name
+                branch = scm.get_ref(EXEC_BRANCH, follow=False)
                 push(
                     dvc,
                     git_remote,