14c08e3da6965eccee804f53ecdf8ecd0c1656f2,tf_agents/utils/nest_utils_test.py,NestedTensorsTest,testGetOuterShapeDynamicShapeBatched,#NestedTensorsTest#,99

Before Change

    self.assertAllEqual(self.evaluate(batch_dim), [7, 5])

  def testGetOuterShapeDynamicShapeBatched(self):
    tensor = tf.placeholder(tf.float32, shape=(None, 1))
    spec = tensor_spec.TensorSpec([1], dtype=tf.float32)
    batch_size = nest_utils.get_outer_shape(tensor, spec)
    with self.cached_session() as sess:
      self.assertEqual(sess.run(batch_size,

After Change

  def testGetOuterShapeDynamicShapeBatched(self):
    spec = tensor_spec.TensorSpec([1], dtype=tf.float32)
    tensor = tf.convert_to_tensor([[0.0]] * 8)
    batch_size = self.evaluate(nest_utils.get_outer_shape(tensor, spec))
    self.assertAllEqual(batch_size, [8])

  def testGetOuterShapeDynamicShapeNotBatched(self):

Instances