729adacfd7e610259e26242e8b214e70ccb32249,tf_agents/experimental/train/utils/replay_buffer_utils_test.py,,,#,71

Before Change

        server_info["test_prioritized_table"].max_size, 333)
    rb.local_server.stop()

if __name__ == "__main__":
  test_utils.main()

After Change

from tf_agents.utils import test_utils

// Lazy loading since not all users have the reverb package installed.
reverb = lazy_loader.LazyLoader("reverb", globals(), "reverb")


class ReplayBufferUtilsTest(test_utils.TestCase):

Instances