openrl.envs.vec_env.vec_info package

Submodules

openrl.envs.vec_env.vec_info.base_vec_info module

class openrl.envs.vec_env.vec_info.base_vec_info.BaseVecInfo(parallel_env_num: int, agent_num: int)

Bases: abc.ABC

abstract append(info: Dict[str, Any]) → None

abstract reset() → None

abstract statistics(buffer: Any) → Dict[str, Any]

openrl.envs.vec_env.vec_info.episode_rewards_info module

class openrl.envs.vec_env.vec_info.episode_rewards_info.EPS_RewardInfo(*args, **kwargs)

Bases: openrl.envs.vec_env.vec_info.simple_vec_info.SimpleVecInfo

statistics(buffer: Any) → Dict[str, Any]

openrl.envs.vec_env.vec_info.nlp_vec_info module

class openrl.envs.vec_env.vec_info.nlp_vec_info.NLPVecInfo(parallel_env_num: int, agent_num: int)

Bases: openrl.envs.vec_env.vec_info.base_vec_info.BaseVecInfo

append(info: Dict[str, Any]) → None

reset() → None

statistics(buffer: Any) → Dict[str, Any]

openrl.envs.vec_env.vec_info.simple_vec_info module

class openrl.envs.vec_env.vec_info.simple_vec_info.SimpleVecInfo(parallel_env_num: int, agent_num: int)

Bases: openrl.envs.vec_env.vec_info.base_vec_info.BaseVecInfo

append(info: Dict[str, Any]) → None

reset() → None

statistics(buffer: Any) → Dict[str, Any]

Module contents

class openrl.envs.vec_env.vec_info.VecInfoFactory

Bases: object

static auto_register(vec_info_class: Any)

static get_vec_info_class(vec_info_class: Any, env: openrl.envs.vec_env.base_venv.BaseVecEnv)

static register(name: str, vec_info: Any)