Skip to content

Commit e7d1bfc

Browse files
committed
fix rewards
1 parent 2cc6c6b commit e7d1bfc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/forge/data/rewards.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,7 @@ def __call__(self, prompt: str, response: str, target: str | None = None) -> flo
167167
"""
168168

169169
# TODO: refactor pending https://github.com/meta-pytorch/torchforge/issues/187
170-
should_debug = debug or (random.random() < self.debug_sample_rate)
170+
should_debug = self.debug or (random.random() < self.debug_sample_rate)
171171

172172
if not response:
173173
if should_debug:

0 commit comments

Comments
 (0)