fix: formatted file

2026-05-04 00:00:05 +02:00 · 2026-05-04 00:00:05 +02:00 · 0a00a978e8
commit 0a00a978e8
parent f66e073f18
1 changed files with 12 additions and 9 deletions
--- a/src/brittle_star_project/trainers/PPOTrainer.py
+++ b/src/brittle_star_project/trainers/PPOTrainer.py
@ -240,15 +240,18 @@ def _rollout_jit(
    truncated_any0 = jnp.zeros((num_envs,), dtype=jnp.bool_)

    (
-        agent_state,
-        episode_stats,
-        next_obs,
-        next_done,
-        key,
-        env_state,
-        terminated_any,
-        truncated_any,
-    ), storage = jax.lax.scan(
+        (
+            agent_state,
+            episode_stats,
+            next_obs,
+            next_done,
+            key,
+            env_state,
+            terminated_any,
+            truncated_any,
+        ),
+        storage,
+    ) = jax.lax.scan(
        partial(
            _step_once,
            sensor=sensor,