sinergym.utils.wrappers.DiscretizeEnv
- class sinergym.utils.wrappers.DiscretizeEnv(env: EplusEnv, discrete_space: Discrete | MultiDiscrete | MultiBinary, action_mapping: Callable[[int | List[int]], float | List[float]])
Wrapper to discretize an action space.
- __init__(env: EplusEnv, discrete_space: Discrete | MultiDiscrete | MultiBinary, action_mapping: Callable[[int | List[int]], float | List[float]])
Wrapper for Discretize action space.
- Parameters:
env (EplusEnv) – Original environment.
discrete_space (Union[gym.spaces.Discrete, gym.spaces.MultiDiscrete, gym.spaces.MultiBinary]) – Discrete Space.
action_mapping (Callable[[Union[int, List[int]]], Union[float, List[float]]]) – Function with action as argument, its output must match with original env action space, otherwise an error will be raised.
Methods
__init__(env, discrete_space, action_mapping)Wrapper for Discretize action space.
action(action)Returns a modified action before
step()is called.class_name()Returns the class name of the wrapper.
close()Closes the wrapper and
env.get_wrapper_attr(name)Gets an attribute from the wrapper and lower environments if name doesn't exist in this object.
has_wrapper_attr(name)Checks if the given attribute is within the wrapper or its environment.
render()Uses the
render()of theenvthat can be overwritten to change the returned data.reset(*[, seed, options])Uses the
reset()of theenvthat can be overwritten to change the returned data.set_wrapper_attr(name, value, *[, force])Sets an attribute on this wrapper or lower environment if name is already defined.
step(action)Runs the
envenv.step()using the modifiedactionfromself.action().wrapper_spec(**kwargs)Generates a WrapperSpec for the wrappers.
Attributes
action_spaceReturn the
Envaction_spaceunless overwritten then the wrapperaction_spaceis used.metadataReturns the
Envmetadata.np_randomReturns the
Envnp_randomattribute.np_random_seedReturns the base environment's
np_random_seed.observation_spaceReturn the
Envobservation_spaceunless overwritten then the wrapperobservation_spaceis used.render_modeReturns the
Envrender_mode.specReturns the
Envspecattribute with the WrapperSpec if the wrapper inherits from EzPickle.unwrappedReturns the base environment of the wrapper.
- action(action: int | List[int])
Returns a modified action before
step()is called.- Parameters:
action – The original
step()actions- Returns:
The modified actions
- property is_discrete: bool
- logger = <Logger WRAPPER DiscretizeEnv (INFO)>