src.environment.parallelenv.worker.subproc

Module Contents

Classes

ShArray	Wrapper of multiprocessing Array.
SubprocEnvWorker	Subprocess worker used in SubprocVectorEnv and ShmemVectorEnv.

Functions

_setup_buf
_worker

Data

_NP_TO_CT

API

src.environment.parallelenv.worker.subproc._NP_TO_CT

None

class src.environment.parallelenv.worker.subproc.ShArray(dtype: numpy.generic, shape: Tuple[int])

Wrapper of multiprocessing Array.

Initialization

save(ndarray: numpy.ndarray) → None

get() → numpy.ndarray

src.environment.parallelenv.worker.subproc._setup_buf(space: gym.Space) → Union[dict, tuple, src.environment.parallelenv.worker.subproc.ShArray]

src.environment.parallelenv.worker.subproc._worker(parent: multiprocessing.connection.Connection, p: multiprocessing.connection.Connection, env_fn_wrapper: src.environment.parallelenv.utils.CloudpickleWrapper, obs_bufs: Optional[Union[dict, tuple, src.environment.parallelenv.worker.subproc.ShArray]] = None) → None

class src.environment.parallelenv.worker.subproc.SubprocEnvWorker(env_fn: Callable[[], gym.Env], share_memory: bool = False, no_warning=False)

Bases: src.environment.parallelenv.worker.base.EnvWorker

Subprocess worker used in SubprocVectorEnv and ShmemVectorEnv.

Initialization

customized_method(func: str, data=None) → Any

get_env_attr(key: str) → Any

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

get_env_obj()

_decode_obs() → Union[dict, tuple, numpy.ndarray]

static wait(workers: List[src.environment.parallelenv.worker.subproc.SubprocEnvWorker], wait_num: int, timeout: Optional[float] = None) → List[src.environment.parallelenv.worker.subproc.SubprocEnvWorker]

send(action: Optional[numpy.ndarray]) → None

send_reset() → None

recv() → Union[Tuple[numpy.ndarray, numpy.ndarray, numpy.ndarray, numpy.ndarray], numpy.ndarray]

seed(seed: Optional[int] = None) → Optional[List[int]]

render(**kwargs: Any) → Any

close_env() → None