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

fix: formatted file

Browse source
This commit is contained in:
Jona Reynaert 2026-05-04 00:00:05 +02:00
parent f66e073f18
commit 0a00a978e8
1 changed files with 12 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/brittle_star_project/trainers/PPOTrainer.py
View file

@ -239,6 +239,7 @@ def _rollout_jit(
terminated_any0 = jnp.zeros((num_envs,), dtype=jnp.bool_) terminated_any0 = jnp.zeros((num_envs,), dtype=jnp.bool_)
truncated_any0 = jnp.zeros((num_envs,), dtype=jnp.bool_) truncated_any0 = jnp.zeros((num_envs,), dtype=jnp.bool_)
(
( (
agent_state, agent_state,
episode_stats, episode_stats,
@ -248,7 +249,9 @@ def _rollout_jit(
env_state, env_state,
terminated_any, terminated_any,
truncated_any, truncated_any,
), storage = jax.lax.scan( ),
storage,
) = jax.lax.scan(
partial( partial(
_step_once, _step_once,
sensor=sensor, sensor=sensor,