typing: executor: use queue.Queue instead of multiprocessing.Queue

iterative · Jun 13, 2022 · 8aeb79c · 8aeb79c
1 parent c368bb7
commit 8aeb79c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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