From cc1ff18de0947cadaa905f65cec735762c21144a Mon Sep 17 00:00:00 2001 From: Tibo De Peuter Date: Tue, 12 May 2026 19:16:38 +0200 Subject: [PATCH] fix: typos --- src/brittle_star_project/environment/obs_processing.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/brittle_star_project/environment/obs_processing.py b/src/brittle_star_project/environment/obs_processing.py index f38d52b..e169ee3 100644 --- a/src/brittle_star_project/environment/obs_processing.py +++ b/src/brittle_star_project/environment/obs_processing.py @@ -109,6 +109,7 @@ def create_obs_processor( joints_per_segment = 2 joints_per_arm = segs_per_arm * joints_per_segment for key, arr in obs.items(): + arr = jnp.asarray(arr) if arr.size == 0: continue @@ -121,7 +122,7 @@ def create_obs_processor( idx = segment_indices[i] taken = jnp.take(arr, idx, axis=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)) arr = jnp.stack(per_agent) @@ -159,7 +160,7 @@ def create_obs_processor( """ values = [] - for key in ordered_keys: + for key in sorted(ordered_keys): if key not in obs: continue