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

fix(PR#41): applied comments for PR #41 review

Browse source
This commit is contained in:
Robin Meersman 2026-04-18 11:07:01 +02:00
parent c880285dc2
commit 34e8ba91c2
4 changed files with 2 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/brittle_star_project/environment/BrittleStarJaxEnvWrapper.py
View file

@ -27,7 +27,7 @@ class BrittleStarJaxEnvWrapper:
)
# Pre-compute masks for observation padding
self._padding_masks = compute_padding_masks(self._morphology.segments_per_arm, (4, 4))
self._padding_masks = compute_padding_masks(self._morphology.segments_per_arm)
self._vectorized_reset = jax.jit(jax.vmap(self._env.reset))
self._vectorized_step = jax.jit(jax.vmap(self._env.step))