fix(testing): target direction updated to work with dim >= 2

src/brittle_star_project/environment/obs_processing.py

@@ -105,6 +105,7 @@ def create_obs_processor(
             # -------- SPLIT TO SEGMENTS --------
             if key in _JOINT_SCALED_KEYS:
                 if morph_mode == MorphMode.SEGMENT:
+                    # TODO: remove magic constants
                     center_size = num_arms * 3 * 2
                     v_center = v[:center_size].reshape(num_arms, 3 * 2)  # (arms, 6)
                     v_segs = v[center_size:].reshape(-1, 2)  # (segs, 2)