dacbench.wrappers.state_tracking_wrapper

Module Contents

Classes

StateTrackingWrapper

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.Wrapper

Wrapper 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