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

fix: typos

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

5
src/brittle_star_project/environment/obs_processing.py
View file

@ -109,6 +109,7 @@ def create_obs_processor(
joints_per_segment = 2 joints_per_segment = 2
joints_per_arm = segs_per_arm * joints_per_segment joints_per_arm = segs_per_arm * joints_per_segment
for key, arr in obs.items(): for key, arr in obs.items():
arr = jnp.asarray(arr)
if arr.size == 0: if arr.size == 0:
continue continue
@ -121,7 +122,7 @@ def create_obs_processor(
idx = segment_indices[i] idx = segment_indices[i]
taken = jnp.take(arr, idx, axis=0) taken = jnp.take(arr, idx, axis=0)
pad_len = segs_per_arm - taken.shape[0] pad_len = segs_per_arm - taken.shape[0]
padded = jnp.pad(taken, [(9, pad_len)] + [(0, 0)] * (taken.ndim - 1)) padded = jnp.pad(taken, [(0, pad_len)] + [(0, 0)] * (taken.ndim - 1))
per_agent.append(padded.reshape(-1)) per_agent.append(padded.reshape(-1))
arr = jnp.stack(per_agent) arr = jnp.stack(per_agent)
@ -159,7 +160,7 @@ def create_obs_processor(
""" """
values = [] values = []
for key in ordered_keys: for key in sorted(ordered_keys):
if key not in obs: if key not in obs:
continue continue