dacbench.wrappers.state_tracking_wrapper
Module Contents
Classes
Wrapper to track state changed over time |
- dacbench.wrappers.state_tracking_wrapper.current_palette
- class dacbench.wrappers.state_tracking_wrapper.StateTrackingWrapper(env, state_interval=None, logger=None)
Bases:
gym.WrapperWrapper to track state changed over time Includes interval mode that returns states in lists of len(interval) instead of one long list.
- __setattr__(self, name, value)
Set attribute in wrapper if available and in env if not
- Parameters
name (str) – Attribute to set
value – Value to set attribute to
- __getattribute__(self, name)
Get attribute value of wrapper if available and of env if not
- Parameters
name (str) – Attribute to get
- Returns
Value of given name
- Return type
value
- reset(self)
Reset environment and record starting state
- Returns
state
- Return type
np.array
- step(self, action)
Execute environment step and record state
- Parameters
action (int) – action to execute
- Returns
state, reward, done, metainfo
- Return type
np.array, float, bool, dict
- get_states(self)
Get state progression
- Returns
all states or all states and interval sorted states
- Return type
np.array or np.array, np.array
- render_state_tracking(self)
Render state progression
- Returns
RBG data of state tracking
- Return type
np.array