File size: 826 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import multiprocessing as mp
import weakref
from typing import Any, Iterator, List, Optional, TypeVar
import zmq
T = TypeVar('T')
the_protocol: int
all_pids: weakref.WeakSet
class EOF:
index: int
def __init__(self, **kw: Any) -> None: ...
def reader(dataset: Any, sockname: str, index: int, num_workers: int) -> None: ...
class MultiLoader:
dataset: Any
workers: int
verbose: bool
pids: List[Optional[mp.Process]]
socket: Optional[zmq.Socket]
ctx: zmq.Context
nokill: bool
prefix: str
sockname: str
def __init__(
self,
dataset: Any,
workers: int = 4,
verbose: bool = False,
nokill: bool = False,
prefix: str = "/tmp/_multi-"
) -> None: ...
def kill(self) -> None: ...
def __iter__(self) -> Iterator[Any]: ...
|