diff --git a/dvc/repo/experiments/executor/base.py b/dvc/repo/experiments/executor/base.py index f4d23a5411..ac3f78e13e 100644 --- a/dvc/repo/experiments/executor/base.py +++ b/dvc/repo/experiments/executor/base.py @@ -39,7 +39,7 @@ ) if TYPE_CHECKING: - from multiprocessing import Queue + from queue import Queue from scmrepo.git import Git diff --git a/dvc/repo/experiments/executor/manager/base.py b/dvc/repo/experiments/executor/manager/base.py index af0e169913..2f53802fe6 100644 --- a/dvc/repo/experiments/executor/manager/base.py +++ b/dvc/repo/experiments/executor/manager/base.py @@ -183,7 +183,7 @@ def _exec_attached(self, repo: "Repo", jobs: Optional[int] = 1): executor.reproduce, info=executor.info, rev=rev, - queue=pid_q, # type: ignore[arg-type] + queue=pid_q, infofile=infofile, log_level=logger.getEffectiveLevel(), ) diff --git a/dvc/repo/experiments/executor/ssh.py b/dvc/repo/experiments/executor/ssh.py index 0d03a6f622..6d96480fc5 100644 --- a/dvc/repo/experiments/executor/ssh.py +++ b/dvc/repo/experiments/executor/ssh.py @@ -19,7 +19,7 @@ from .base import BaseExecutor, ExecutorInfo, ExecutorResult if TYPE_CHECKING: - from multiprocessing import Queue + from queue import Queue from scmrepo.git import Git