The reward model scores whole segments of behaviour, not single steps. A segment's return is the sum of the per-step predictions, and comparisons are made between segment returns.
Implement segment_return(rewards), returning the sum of the predicted rewards.
Sum the array.
No results yet
Run your code to check it against the examples, or submit to run every test.