Skip to content

[Integrations] Support PyTorch OpenEnv#543

Merged
SumanthRH merged 23 commits intoNovaSky-AI:mainfrom
lynnliu030:openenv
Oct 23, 2025
Merged

[Integrations] Support PyTorch OpenEnv#543
SumanthRH merged 23 commits intoNovaSky-AI:mainfrom
lynnliu030:openenv

Conversation

@lynnliu030
Copy link
Member

@lynnliu030 lynnliu030 commented Oct 21, 2025

Integrate Pytorch OpenEnv with SkyRL, register it as a SkyRL-Gym environment and use SkyRLGymGenerator.

  • Support environment such as coding, echo, finrl, openspiel, atari, and sumo RL.
  • Provide example training scripts (with dummy train / validation data) for coding and echo environment

Main integration is in integrations/openenv/env.py

Example reward:
image

Copy link
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces support for PyTorch OpenEnv, including adding the openenv dependency to pyproject.toml, registering the environment in __init__.py, creating the OpenEnv class in env.py, adding an installation script install_environment.py, creating tests in test_openenv.py, and adding a dummy dataset example in dummy_openenv_dataset.py with a corresponding run script run_dummy_openenv.sh. The litellm dependency version is also updated in pyproject.toml.

@lynnliu030 lynnliu030 requested a review from SumanthRH October 22, 2025 18:07
@lynnliu030 lynnliu030 changed the title [gym] Support PyTorch OpenEnv [Integrations] Support PyTorch OpenEnv Oct 22, 2025
lynnliu030 and others added 2 commits October 22, 2025 19:07
Signed-off-by: lynnliu030 <lshu@berkeley.edu>
Co-authored-by: Sumanth Hegde <sumanthrh@anyscale.com>
Signed-off-by: lynnliu030 <lshu@berkeley.edu>
lynnliu030 and others added 8 commits October 22, 2025 19:23
x
Signed-off-by: lynnliu030 <lshu@berkeley.edu>
x
Signed-off-by: lynnliu030 <lshu@berkeley.edu>
Co-authored-by: Sumanth R Hegde <39546518+SumanthRH@users.noreply.github.com>
lynnliu030 and others added 3 commits October 23, 2025 17:49
Signed-off-by: lynnliu030 <lshu@berkeley.edu>
Co-authored-by: Sumanth Hegde <sumanthrh@anyscale.com>
Copy link
Member

@SumanthRH SumanthRH left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LFG

@SumanthRH SumanthRH merged commit 7813316 into NovaSky-AI:main Oct 23, 2025
3 checks passed
atemaguer pushed a commit to atemaguer/SkyRL that referenced this pull request Oct 24, 2025
Integrate [Pytorch OpenEnv](https://github.com/meta-pytorch/OpenEnv)
with SkyRL, register it as a SkyRL-Gym environment and use
SkyRLGymGenerator.

* Support environment such as coding, echo, finrl, openspiel, atari, and
sumo RL.
* Provide example training scripts (with dummy train / validation data)
for coding and echo environment

Main integration is in `integrations/openenv/env.py` 

Example reward:
<img width="1554" height="826" alt="image"
src="https://github.com/user-attachments/assets/bfe6251c-bb21-41bd-8e7f-70eeb42fd376"
/>

---------

Signed-off-by: lynnliu030 <lshu@berkeley.edu>
Co-authored-by: Sumanth Hegde <sumanthrh@anyscale.com>
Co-authored-by: Sumanth R Hegde <39546518+SumanthRH@users.noreply.github.com>
li-boxuan pushed a commit to li-boxuan/SkyRL that referenced this pull request Nov 23, 2025
Integrate [Pytorch OpenEnv](https://github.com/meta-pytorch/OpenEnv)
with SkyRL, register it as a SkyRL-Gym environment and use
SkyRLGymGenerator.

* Support environment such as coding, echo, finrl, openspiel, atari, and
sumo RL.
* Provide example training scripts (with dummy train / validation data)
for coding and echo environment

Main integration is in `integrations/openenv/env.py` 

Example reward:
<img width="1554" height="826" alt="image"
src="https://github.com/user-attachments/assets/bfe6251c-bb21-41bd-8e7f-70eeb42fd376"
/>

---------

Signed-off-by: lynnliu030 <lshu@berkeley.edu>
Co-authored-by: Sumanth Hegde <sumanthrh@anyscale.com>
Co-authored-by: Sumanth R Hegde <39546518+SumanthRH@users.noreply.github.com>
dzorlu pushed a commit to fleet-ai/SkyRL that referenced this pull request Feb 4, 2026
Integrate [Pytorch OpenEnv](https://github.com/meta-pytorch/OpenEnv)
with SkyRL, register it as a SkyRL-Gym environment and use
SkyRLGymGenerator.

* Support environment such as coding, echo, finrl, openspiel, atari, and
sumo RL.
* Provide example training scripts (with dummy train / validation data)
for coding and echo environment

Main integration is in `integrations/openenv/env.py` 

Example reward:
<img width="1554" height="826" alt="image"
src="https://github.com/user-attachments/assets/bfe6251c-bb21-41bd-8e7f-70eeb42fd376"
/>

---------

Signed-off-by: lynnliu030 <lshu@berkeley.edu>
Co-authored-by: Sumanth Hegde <sumanthrh@anyscale.com>
Co-authored-by: Sumanth R Hegde <39546518+SumanthRH@users.noreply.github.com>
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.

2 participants