feat: split n flat n pad decentered

2026-05-06 18:52:19 +00:00 · 2026-05-06 18:52:19 +00:00 · 9c20c28b2a
commit 9c20c28b2a
parent 5e7fd27b61
5 changed files with 88 additions and 50 deletions
--- a/tests/test_network_shapes.py
+++ b/tests/test_network_shapes.py
@ -32,6 +32,7 @@ def test_centralized_forward_pass_with_padding():
        num_arms=num_arms,
        padding_masks=masks,
        morph_mode=MorphMode.CENTRALIZED,
+        segments_per_arm=segments_per_arm,
    )
    global_state = obs_processor(amputated_obs)