Fix the off-by-one error in R2D2 state storage.
Fix the off-by-one error in R2D2 state storage.

PiperOrigin-RevId: 469736397
Change-Id: I4394cde051a0b3562c9bb72d211a35fed0a11b91

deepmind/acme,deepmind/acme