def testRetraceThreeTimeSteps(self):
    """Subject Retrace to a two-sequence, three-timestep minibatch."""
    retrace = retrace_ops.retrace(
        self.lambda_, self.qs, self.targnet_qs, self.actions, self.rewards,
        self.pcontinues, self.target_policy_probs, self.behaviour_policy_probs)

    with self.test_session() as sess:
      self.assertAllClose(sess.run(retrace.loss), self.expected_result)
Subject Retrace to a two-sequence, three-timestep minibatch.
trfl/retrace_ops_test.py