Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix gymnasium version as pre-1.0.0 #471

Merged
merged 4 commits into from
Oct 18, 2024

Conversation

zimka
Copy link
Contributor

@zimka zimka commented Oct 13, 2024

What this does

This PR fixes #470 (🐛 Bug)
The issue is caused by recent v1.0.0 release of gymnasium from 08.10.2024.

It looks like the final_info key-value pair is no longer available in the output of SyncVectorEnv.step, so evaluation code for success_rate in lerobot/scripts/eval.py gets broken.

The proper fix of the issue requires changes in eval.py and subsequent checks versus available configs.
This PR fixes evaluation behavior "as it was before" and postpones the usage of gymnasium==1.0.0 until necessary changes and checks are done.

How it was tested

I have installed gymnasium==0.29.1 (which is the latest pre-1.0.0) locally and re-evaluated trained diffusion policy for pushT and achieved correct pc_success:
{'avg_sum_reward': 101.66934858131904, 'avg_max_reward': 0.9727046033094218, 'pc_success': 76.0, 'eval_s': 69.3367211818695, 'eval_ep_s': 1.3867344379425048}

How to checkout & try? (for the reviewer)

git remote add zimka https://github.com/zimka/lerobot
git checkout zimka fix_gymnasium_version

pyproject.toml Show resolved Hide resolved
@Cadene Cadene merged commit c351e1f into huggingface:main Oct 18, 2024
6 checks passed
astroyat pushed a commit to astroyat/lerobot that referenced this pull request Nov 23, 2024
Co-authored-by: Remi <re.cadene@gmail.com>
Co-authored-by: Remi <remi.cadene@huggingface.co>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
2 participants