-
Notifications
You must be signed in to change notification settings - Fork 312
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
[Feature] Make advantages compatible with Terminated, Truncated, Done #1581
Conversation
# Conflicts: # test/test_libs.py # torchrl/envs/gym_like.py
Co-authored-by: Skander Moalla <[email protected]>
Co-authored-by: Skander Moalla <[email protected]>
Co-authored-by: Skander Moalla <[email protected]>
Could you merge main? |
You should be able to do it |
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 this!
Since there are so many changes, I ll do another pass when main is merged,
but for now here are some comments
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.
LGTM
…pytorch#1581) Co-authored-by: Skander Moalla <[email protected]> Co-authored-by: Matteo Bettini <[email protected]>
I manually tested that everything was working without truncated key for bc compatibility.
Not sure if we should test that things work if terminated is missing in all losses as it's very redundant and the cost tests are already very long.
I didn't include the truncated -> done fallback in the docstrings, maybe we should but I don't want to overload them.
Depends on #1539