Skip to content

Expose Same-Step terminal observations#6219

Open
ooctipus wants to merge 1 commit into
isaac-sim:developfrom
ooctipus:support-vector-autoreset-modes
Open

Expose Same-Step terminal observations#6219
ooctipus wants to merge 1 commit into
isaac-sim:developfrom
ooctipus:support-vector-autoreset-modes

Conversation

@ooctipus

@ooctipus ooctipus commented Jun 21, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Preserve Isaac Lab existing Same-Step autoreset behavior without adding autoreset mode config.
  • Declare metadata["autoreset_mode"] = gym.vector.AutoresetMode.SAME_STEP on RL envs.
  • Capture terminal observations before Same-Step reset and expose them through extras["final_obs"] for Direct RL, Manager-based RL, and Direct MARL envs.
  • Update Sb3VecEnvWrapper to use extras["final_obs"] as SB3 terminal_observation when envs reset on the same step.

Validation

  • pre-commit run --files source/isaaclab/changelog.d/support-autoreset-modes.rst source/isaaclab/isaaclab/envs/direct_marl_env.py source/isaaclab/isaaclab/envs/direct_rl_env.py source/isaaclab/isaaclab/envs/manager_based_rl_env.py source/isaaclab_rl/changelog.d/support-autoreset-final-obs.rst source/isaaclab_rl/isaaclab_rl/sb3.py source/isaaclab_rl/test/test_sb3_wrapper_autoreset.py
  • PYTHONPATH=source/isaaclab:source/isaaclab_rl /home/zhengyuz/Projects/IsaacLab.wt/develop/.venv/bin/python -m pytest source/isaaclab_rl/test/test_sb3_wrapper_autoreset.py
  • PYTHONPATH=source/isaaclab:source/isaaclab_rl /home/zhengyuz/Projects/IsaacLab.wt/develop/.venv/bin/python -m py_compile source/isaaclab/isaaclab/envs/direct_marl_env.py source/isaaclab/isaaclab/envs/direct_rl_env.py source/isaaclab/isaaclab/envs/manager_based_rl_env.py source/isaaclab_rl/isaaclab_rl/sb3.py source/isaaclab_rl/test/test_sb3_wrapper_autoreset.py
  • git diff --check

@github-actions github-actions Bot added the isaac-lab Related to Isaac Lab team label Jun 21, 2026
@ooctipus ooctipus force-pushed the support-vector-autoreset-modes branch 5 times, most recently from 60bc064 to 9df17c3 Compare June 21, 2026 06:04
@greptile-apps

greptile-apps Bot commented Jun 21, 2026

Copy link
Copy Markdown
Contributor

Greptile encountered an error while reviewing this PR. Please reach out to support@greptile.com for assistance.

@ooctipus ooctipus force-pushed the support-vector-autoreset-modes branch 3 times, most recently from 1de310c to a48f14d Compare June 21, 2026 09:19
@ooctipus ooctipus changed the title Add configurable RL autoreset modes Expose Same-Step terminal observations Jun 21, 2026
@ooctipus ooctipus force-pushed the support-vector-autoreset-modes branch from a48f14d to 9b0e1d0 Compare June 21, 2026 09:28
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

isaac-lab Related to Isaac Lab team

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant