Skip to content

Commit 275a9f2

Browse files
committed
Replace mp queue import with correct queue.Queue
Signed-off-by: Samuel Monson <[email protected]>
1 parent 4b0bf0b commit 275a9f2

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

src/guidellm/scheduler/result.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from dataclasses import dataclass
2-
from multiprocessing import Queue
2+
from queue import Queue
33
from typing import (
44
Generic,
55
Literal,

src/guidellm/scheduler/scheduler.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,9 @@
33
import time
44
from collections.abc import AsyncGenerator, Iterable, Iterator
55
from concurrent.futures import ProcessPoolExecutor
6-
from multiprocessing import Manager, Queue
6+
from multiprocessing import Manager
77
from queue import Empty as QueueEmpty
8+
from queue import Queue
89
from typing import (
910
Any,
1011
Generic,

src/guidellm/scheduler/worker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@
44
from abc import ABC, abstractmethod
55
from collections.abc import AsyncGenerator
66
from dataclasses import dataclass
7-
from multiprocessing import Queue
87
from queue import Empty as QueueEmpty
8+
from queue import Queue
99
from typing import (
1010
Any,
1111
Generic,

0 commit comments

Comments
 (0)