From 8aeb79ca70fec0573ee751d806fc050638c55e86 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Saugat=20Pachhai=20=28=E0=A4=B8=E0=A5=8C=E0=A4=97=E0=A4=BE?= =?UTF-8?q?=E0=A4=A4=29?= Date: Mon, 13 Jun 2022 21:27:01 +0545 Subject: [PATCH] typing: executor: use queue.Queue instead of multiprocessing.Queue --- dvc/repo/experiments/executor/base.py | 2 +- dvc/repo/experiments/executor/manager/base.py | 2 +- dvc/repo/experiments/executor/ssh.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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