9fc32f79c4d5968274cca37a29cf6611f4808a8c,tf_agents/utils/common_test.py,IndexWithActionsTest,testTwoOuterDimsUnknownShape,#IndexWithActionsTest#,198

Before Change

    q_values = tf.placeholder(tf.float32, shape=[None, None, None])
    actions = tf.placeholder(tf.int32, shape=[None, None])
    values = common.index_with_actions(q_values, actions)
    with self.cached_session() as sess:
      self.assertAllClose(
          [[51, 52]],
          sess.run(values,
                   feed_dict={q_values: [[[50, 51], [52, 53]]],
                              actions: [[1, 0]]}))


class PeriodicallyTest(tf.test.TestCase):
  Tests function periodically.

  @test_util.run_in_graph_and_eager_modes()

After Change

  def testTwoOuterDimsUnknownShape(self):
    q_values = tf.convert_to_tensor(np.array([[[50, 51], [52, 53]]],
                                             dtype=np.float32))
    actions = tf.convert_to_tensor(np.array([[1, 0]], dtype=np.int32))
    values = common.index_with_actions(q_values, actions)

    self.assertAllClose([[51, 52]], self.evaluate(values))

Instances