Skip to content

Add double pendulum environment#550

Merged
jsuarez5341 merged 1 commit into
PufferAI:4.0from
JackB-4:add-double-pendulum-env
May 13, 2026
Merged

Add double pendulum environment#550
jsuarez5341 merged 1 commit into
PufferAI:4.0from
JackB-4:add-double-pendulum-env

Conversation

@JackB-4
Copy link
Copy Markdown

@JackB-4 JackB-4 commented May 1, 2026

Adds double_pendulum, a native Ocean control environment.

Details

Environment

Box(8,) observations: normalized cart position/velocity + sin/cos angle and angular velocity for both pendulum links
Discrete(3) actions: left force, coast, right force
Two-link pendulum on a cart, starting from the downward hanging position
Reward: 0.5 * height + hold_bonus, where height measures tip height and hold_bonus rewards sustained stable upright balance
Terminates on cart out-of-bounds, invalid physics, or timeout

On my machine, the env trains around ~1M SPS with the included config.

double_pendulum_demo.mp4

@jsuarez5341 jsuarez5341 merged commit 25b3165 into PufferAI:4.0 May 13, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants