Fix index out of range error in q_rnn_network_test.
Fix index out of range error in q_rnn_network_test.

PiperOrigin-RevId: 234610588
Change-Id: Ic994f6e585ec54017aadf5f0c3c52fcecbca5fcc

tensorflow/agents,tensorflow/agents