subproc

Source code: tianshou/env/worker/subproc.py

---

class ShArray(dtype: generic, shape: tuple[int])

Wrapper of multiprocessing Array.

get() → ndarray

save(ndarray: ndarray) → None

class SubprocEnvWorker(env_fn: Callable[[], Env], share_memory: bool = False)

Subprocess worker used in SubprocVectorEnv and ShmemVectorEnv.

close_env() → None

get_env_attr(key: str) → Any

recv() → tuple[numpy.ndarray, numpy.ndarray, numpy.ndarray, numpy.ndarray, numpy.ndarray] | tuple[numpy.ndarray, dict]

Receive result from low-level worker.

If the last “send” function sends a NULL action, it only returns a single observation; otherwise it returns a tuple of (obs, rew, done, info) or (obs, rew, terminated, truncated, info), based on whether the environment is using the old step API or the new one.

render(**kwargs: Any) → Any

Render the environment.

reset(**kwargs: Any) → tuple[numpy.ndarray, dict]

seed(seed: int | None = None) → list[int] | None

send(action: numpy.ndarray | None, **kwargs: Any) → None

Send action signal to low-level worker.

When action is None, it indicates sending “reset” signal; otherwise it indicates “step” signal. The paired return value from “recv” function is determined by such kind of different signal.

set_env_attr(key: str, value: Any) → None

static wait(workers: list[SubprocEnvWorker], wait_num: int, timeout: float | None = None) → list[SubprocEnvWorker]

Given a list of workers, return those ready ones.