Visualization

Available Visualizations:

Visualization Base Class

class gym_electric_motor.core.ElectricMotorVisualization[source]

Bases: Callback

Base class for all visualizations in GEM.

Notes

This class extends Callback by adding render() to update the user interface. Data is transferred via the inherited callback hooks (e.g., on_step_end()) and rendered inside render().

on_close()

Gets called at the beginning of a close

on_reset_begin()

Gets called at the beginning of each reset

on_reset_end(state, reference)

Gets called at the end of each reset

on_step_begin(k, action)

Gets called at the beginning of each step

on_step_end(k, state, reference, reward, terminated)

Gets called at the end of each step

render()[source]

Update the visualization UI.

set_env(env)

Sets the environment of the motor.