feat(rldb): heterogeneous u_socket action support - #566
Draft
ElmoPA wants to merge 1 commit into
Draft
Conversation
Adds what a U-socket pusher needs alongside the 2-D circle embodiments: - embodiment: pin PUSHSHAPES_SIM_U_SOCKET = 19. 18 was HUMAN_BIMANUAL under the pre-collapse numbering and the existing u_socket datasets/checkpoints (the bf_*_u3* 2026-08 lineage) already encode 19. - transforms: ThetaToRotVec / RotVecToTheta, so an angle is carried as (cos, sin) and cannot wrap discontinuously during training; the revert is needed because env.step still wants a 3-dim u_socket action. - pushshapes: get_keymap_with_pusher_cmd + the rotvec transform lists. - stages_flow: generalise pusher_hold from hardcoded 2-D xy to a D-dimensional pose (state[:, :Dd]), with a guard that raises when the state carries fewer coordinates than the action, instead of silently padding from the wrong columns. - pl_data_utils: route packed datasets through pack_collate, which emits cu_seqlens, rather than the configured collate_fn.
Contributor
Author
This was referenced Aug 12, 2026
ElmoPA
changed the base branch from
config/organize-hydra
to
graphite-base/566
August 21, 2026 03:19
This was referenced Aug 21, 2026
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.

Adds what a U-socket pusher needs alongside the 2-D circle embodiments:
the pre-collapse numbering and the existing u_socket datasets/checkpoints
(the bf__u3 2026-08 lineage) already encode 19.
(cos, sin) and cannot wrap discontinuously during training; the revert is
needed because env.step still wants a 3-dim u_socket action.
D-dimensional pose (state[:, :Dd]), with a guard that raises when the state
carries fewer coordinates than the action, instead of silently padding from
the wrong columns.
cu_seqlens, rather than the configured collate_fn.