fix: formatted file

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,
+            agent_state,
-        next_obs,
+            episode_stats,
-        next_done,
+            next_obs,
-        key,
+            next_done,
-        env_state,
+            key,
-        terminated_any,
+            env_state,
-        truncated_any,
+            terminated_any,
-    ), storage = jax.lax.scan(
+            truncated_any,
+        ),
+        storage,
+    ) = jax.lax.scan(
         partial(
             _step_once,
             sensor=sensor,