dacbench.wrappers.episode_time_tracker¶
Module Contents¶
Classes¶
Wrapper to track time spent per episode. |
- dacbench.wrappers.episode_time_tracker.current_palette¶
- class dacbench.wrappers.episode_time_tracker.EpisodeTimeWrapper(env, time_interval=None, logger=None)¶
Bases:
gym.WrapperWrapper to track time spent per episode. Includes interval mode that returns times 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
- step(self, action)¶
Execute environment step and record time
- Parameters
action (int) – action to execute
- Returns
state, reward, done, metainfo
- Return type
np.array, float, bool, dict
- get_times(self)¶
Get times
- Returns
all times or all times and interval sorted times
- Return type
np.array or np.array, np.array
- render_step_time(self)¶
Render step times
- render_episode_time(self)¶
Render episode times