lit_sb3.vec_env.vec_extract_dict_obs

source module lit_sb3.vec_env.vec_extract_dict_obs

Classes

• VecExtractDictObs — A vectorized wrapper for extracting dictionary observations.

source class VecExtractDictObs(venv: VecEnv, key: str)

Bases : VecEnvWrapper

A vectorized wrapper for extracting dictionary observations.

Parameters

• venv : VecEnv — The vectorized environment

• key : str — The key of the dictionary observation

Methods

• reset

• step_wait

source method VecExtractDictObs.reset() → np.ndarray

source method VecExtractDictObs.step_wait() → VecEnvStepReturn