def testDpgDqda(self):
    """Tests the gradient qd/qa produced by the DPGLearner is correct."""
    with self.test_session() as sess:
      sess.run(tf.global_variables_initializer())
      value_grad = np.transpose(self.w.eval())[0]
      for i in range(int(self.batch_size)):
        self.assertAllClose(self.dpg_extra.dqda.eval()[i], value_grad)
Tests the gradient qd/qa produced by the DPGLearner is correct.
trfl/dpg_ops_test.py