c611d9b6b35e44326bc18096dc57abe18eb6c90b,tf_agents/replay_buffers/tf_uniform_replay_buffer_test.py,TFUniformReplayBufferTest,testGatherAllOverCapacity,#TFUniformReplayBufferTest#Any#,305

Before Change

        spec, batch_size=batch_size, max_length=10)

    // Each element has its batch index in the 100s place.
    actions = tf.stack([
        tf.Variable(x * 100).count_up_to(15 + x * 100)
        for x in range(batch_size)
    ])

    add_op = replay_buffer.add_batch(actions)
    items = replay_buffer.gather_all()
    expected = [

After Change

    ]

    self.evaluate(tf.compat.v1.global_variables_initializer())
    self.evaluate(add_data())
    items = replay_buffer.gather_all()
    items_ = self.evaluate(items)
    self.assertAllClose(expected, items_)

Instances