Skip to content

[Feature] Add satellite MuJoCo SAC examples#3802

Merged
vmoens merged 1 commit into
pytorch:mainfrom
vmoens:codex/satellite-examples
Jun 2, 2026
Merged

[Feature] Add satellite MuJoCo SAC examples#3802
vmoens merged 1 commit into
pytorch:mainfrom
vmoens:codex/satellite-examples

Conversation

@vmoens

@vmoens vmoens commented May 30, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Adds synchronous SAC + PER training for the satellite MuJoCo task.
  • Adds a fully asynchronous SAC + PER variant for the same task.
  • Adds shared satellite-example helpers for environment construction, evaluation setup, model creation, logging, and deterministic in-memory eval starts.
  • Adds a regression test that checks eval starts are applied before the wrapped environment reset.

Stack status

#3700 has landed and this branch has been rebased onto main. The PR diff now contains only the satellite example files plus the example-specific regression test.

Test plan

  • python3 -m py_compile test/libs/test_mujoco.py examples/satellite/_utils.py examples/satellite/sac_per.py examples/satellite/sac_per_fully_async.py
  • python examples/satellite/sac_per.py
  • python examples/satellite/sac_per_fully_async.py
  • pytest test/libs/test_mujoco.py -k eval_primer -v

@pytorch-bot

pytorch-bot Bot commented May 30, 2026

Copy link
Copy Markdown

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/rl/3802

Note: Links to docs will display an error until the docs builds have been completed.

❌ 6 New Failures, 1 Unrelated Failure

As of commit d6fb99d with merge base 97ae8cc (image):

NEW FAILURES - The following jobs have failed:

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label May 30, 2026
@github-actions github-actions Bot added Feature New feature Documentation Improvements or additions to documentation CI Has to do with CI setup (e.g. wheels & builds, tests...) Environments Adds or modifies an environment wrapper Examples and removed Feature New feature labels May 30, 2026
@vmoens vmoens force-pushed the codex/satellite-examples branch from a7699ec to 2daa05f Compare May 30, 2026 00:20
@github-actions github-actions Bot added the Feature New feature label May 30, 2026
@vmoens vmoens force-pushed the codex/satellite-examples branch from 2daa05f to d6fb99d Compare May 30, 2026 01:38
@vmoens vmoens marked this pull request as ready for review June 2, 2026 11:40
@vmoens vmoens merged commit f3c9cfb into pytorch:main Jun 2, 2026
105 of 114 checks passed
@vmoens vmoens deleted the codex/satellite-examples branch June 2, 2026 11:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CI Has to do with CI setup (e.g. wheels & builds, tests...) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. Documentation Improvements or additions to documentation Environments Adds or modifies an environment wrapper Examples Feature New feature

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant