Skip to content

VecMergeDictActionWrapper

Full path: schola.sb3.utils.VecMergeDictActionWrapper

schola.sb3.utils.VecMergeDictActionWrapper

VecMergeDictActionWrapper

VecMergeDictActionWrapper(venv)

Bases: VecEnvWrapper

A vectorized wrapper for merging a dictionary of actions into 1 single action. All actions in the dictionary must be of compatible types.

Parameters

venv (VecEnv) : The vectorized environment being wrapped.

Methods

ItemDescription
init
reset
step
step_async
step_wait

init

__init__(venv)

Parameters

venv (VecEnv)


reset

reset()

Returns

VecEnvObs


step

step(action)

Parameters

action (ndarray)

Returns

Tuple[VecEnvObs, ndarray, ndarray, List[Dict]]


step_async

step_async(actions)

Parameters

actions (ndarray)

Returns

None


step_wait

step_wait()

Returns

Tuple[VecEnvObs, ndarray, ndarray, List[Dict]]