sinergym.utils.wrappers.BaseLoggerWrapper
- class sinergym.utils.wrappers.BaseLoggerWrapper(*args, **kwargs)
- __init__(*args, **kwargs)
Wraps an environment to allow a modular transformation of the
step()andreset()methods.- Parameters:
env – The environment to wrap
Methods
__init__(*args, **kwargs)Wraps an environment to allow a modular transformation of the
step()andreset()methods.calculate_custom_metrics(obs, action, ...)Calculate custom metrics from current interaction (or passed using self.data_logger attributes)
class_name()Returns the class name of the wrapper.
close()Close the environment and save normalization calibration.
Return data summary for the logger. This method should be implemented in the child classes.
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])Reset the environment and the information logged.
set_wrapper_attr(name, value, *[, force])Sets an attribute on this wrapper or lower environment if name is already defined.
step(action)Uses the
step()of theenvthat can be overwritten to change the returned data.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.
- abstractmethod calculate_custom_metrics(obs: ndarray, action: ndarray, reward: SupportsFloat, info: Dict[str, Any], terminated: bool, truncated: bool)
Calculate custom metrics from current interaction (or passed using self.data_logger attributes)
- Parameters:
obs (np.ndarray) – Observation from environment.
action (np.ndarray) – Action taken in environment.
reward (float) – Reward received from environment.
info (Dict[str, Any]) – Information from environment.
terminated (bool) – Flag to indicate if episode is terminated.
truncated (bool) – Flag to indicate if episode is truncated.
- close()
Close the environment and save normalization calibration.
- abstractmethod get_episode_summary() Dict[str, float]
- Return data summary for the logger. This method should be implemented in the child classes.
This method determines the data summary of episodes in Sinergym environments.
- Returns:
Data summary.
- Return type:
Dict[str, float]
- reset(seed: int | None = None, options: Dict[str, Any] | None = None) Tuple[ndarray, Dict[str, Any]]
Reset the environment and the information logged.