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

Add wrapper to help logging episode returns #7

Merged
merged 4 commits into from Aug 18, 2022
Merged

Conversation

ffelten
Copy link
Collaborator

@ffelten ffelten commented Aug 12, 2022

The implementation is mostly a copy paste of the original gym. I had to copy paste instead of override and call to super because the way the return is a numpy array, which is mutable, and the original implementation resets it to 0. Hence, if we kept the original, the return will always be a vector of zeros (because resetted)

mo_gym/utils.py Show resolved Hide resolved
mo_gym/utils.py Show resolved Hide resolved
@ffelten ffelten merged commit 06f0b9c into main Aug 18, 2022
@ffelten ffelten deleted the feature/episode_wrapper branch August 18, 2022 09:33
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.

None yet

2 participants