-
Notifications
You must be signed in to change notification settings - Fork 113
Change _reward_fn to take action taken #166
Conversation
Hi @RobertKirk! Thank you for your pull request and welcome to our community. Action RequiredIn order to merge any pull request (code, docs, etc.), we require contributors to sign our Contributor License Agreement, and we don't seem to have one on file for you. ProcessIn order for us to review and merge your suggested changes, please sign at https://code.facebook.com/cla. If you are contributing on behalf of someone else (eg your employer), the individual CLA may not be sufficient and your employer may need to sign the corporate CLA. Once the CLA is signed, our tooling will perform checks and validations. Afterwards, the pull request will be tagged with If you have received this in error or have any questions, please contact us at cla@fb.com. Thanks! |
Thank you for signing our Contributor License Agreement. We can now accept your code for this (and any) Facebook open source project. 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.
Looks good, short comment below.
@@ -194,7 +196,7 @@ def __init__(self, *args, **kwargs): | |||
|
|||
super().__init__(*args, options=options, **kwargs) | |||
|
|||
def _reward_fn(self, last_observation, observation, end_status): | |||
def _reward_fn(self, last_observation, action, observation, end_status): | |||
"""Difference between previous gold and new gold.""" | |||
del end_status # Unused |
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! To keep this in line with how this code was written so far, would you mind adding del action
lines to make linters happy about action
not being used in these functions at this point?
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.
Sure thing, I've made that change
In standard RL often the reward function takes as input the action taken. This change allows that to happen by adjusting _reward_fn. See issue facebookresearch#155 for more details
4f404ab
to
59a473e
Compare
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.
Awesome, thanks!
In standard RL often the reward function takes as input the action
taken. This change allows that to happen by adjusting _reward_fn.
See issue #155 for more details
Closes #155