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_target_direction.py
+++ b/tests/test_target_direction.py
@ -53,6 +53,7 @@ def test_processor_converts_to_egocentric_direction():
        num_arms=num_arms,
        padding_masks=env.padding_masks,
        morph_mode=MorphMode.CENTRALIZED,
+        segments_per_arm=segments_per_arm,
    )

    env_state = env.reset(seed=42)