-
Notifications
You must be signed in to change notification settings - Fork 85
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: make autoreset wrapper return 2 on reset #123
fix: make autoreset wrapper return 2 on reset #123
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thank you for your contribution @rodSiry! 🚀 This looks good to me! Could you update the PR title to follow the conventional commit style format (see our contribution guidelines) and to be more informative? For example: fix: make autoreset wrapper return 2 on reset
.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi, thanks for the contribution. Would you mind applying the same fix to the VmapAutoResetWrapper
? By saying so, I realize we should have shared some code between the two...
Thanks again!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good to me! One last thing I noticed: we need to update the class docstring of AutoResetWrapper
to account for not resetting the step_type to FIRST anymore. Would you mind doing so? Thanks!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the fix!
Fixed the bad autoreset behavior described in issue #106
Now, wrapped environment returns step_type = 2 when resetting.
Adapted corresponding test function, with assert step_type == 2.