From 9c9ca39fb4cb6718e47b2650e0c2da47445bca2e Mon Sep 17 00:00:00 2001 From: Peter Rowlands Date: Tue, 23 May 2023 15:47:35 +0900 Subject: [PATCH] exp: fix rebased imports --- dvc/repo/experiments/executor/base.py | 1 + dvc/repo/experiments/queue/base.py | 1 + 2 files changed, 2 insertions(+) diff --git a/dvc/repo/experiments/executor/base.py b/dvc/repo/experiments/executor/base.py index f7e363963..47311acd3 100644 --- a/dvc/repo/experiments/executor/base.py +++ b/dvc/repo/experiments/executor/base.py @@ -461,6 +461,7 @@ class BaseExecutor(ABC): from dvc.repo.checkout import checkout as dvc_checkout from dvc.repo.reproduce import reproduce as dvc_reproduce from dvc.stage import PipelineStage + from dvc.ui import ui auto_push = env2bool(DVC_EXP_AUTO_PUSH) git_remote = os.getenv(DVC_EXP_GIT_REMOTE, None) diff --git a/dvc/repo/experiments/queue/base.py b/dvc/repo/experiments/queue/base.py index a307f939d..452cba2a7 100644 --- a/dvc/repo/experiments/queue/base.py +++ b/dvc/repo/experiments/queue/base.py @@ -705,6 +705,7 @@ class BaseStashQueue(ABC): def active_repo(self, name: str) -> "Repo": """Return a Repo for the specified active experiment if it exists.""" + from dvc.exceptions import DvcException from dvc.repo import Repo from dvc.repo.experiments.exceptions import ( ExpNotStartedError,