tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
Archived
1
Fork 0
Code Issues Releases Activity

fix(evaluate): add missing value to video

Browse source
This commit is contained in:
Tibo De Peuter 2026-05-12 19:15:44 +02:00
parent a8d0b4797a
commit abf8a3b3e9
Signed by: tdpeuter
SSH key fingerprint: SHA256:u/h/LVoqKF1Iz02uOyxe6hcjmoZASCGV2HM0TG9ZMoU
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/brittle_star_project/evaluation/video.py
View file

@ -97,10 +97,10 @@ def record_episode(
data = state.mj_data data = state.mj_data
renderer = mujoco.Renderer(model, width=width, height=height) renderer = mujoco.Renderer(model, width=width, height=height)
ep_return = 0.0 ep_return = 0.0
observations = _get_observations(state) observations = _get_observations(state)
prev_dist = _get_xy_distance_to_target(observations) if observations else None prev_dist = _get_xy_distance_to_target(observations) if observations else None
initial_dist = prev_dist
reached_target = _target_reached(state=state) reached_target = _target_reached(state=state)
frames = [] frames = []
@ -145,4 +145,5 @@ def record_episode(
length=steps, length=steps,
reached_target=reached_target, reached_target=reached_target,
final_xy_dist=final_dist, final_xy_dist=final_dist,
initial_target_distance=initial_dist,
) )