Merge pull request #897 from cmarlin:init_replay_buffers

copybara-github · copybara-github · commit c8a7175529f3 · 2023-12-14T13:10:22.000-08:00
PiperOrigin-RevId: 591030872
Change-Id: Ibad67a777e4fe9e41ef61e239b5cca3e8bf0b5fa
diff --git a/tf_agents/replay_buffers/__init__.py b/tf_agents/replay_buffers/__init__.py
@@ -15,6 +15,7 @@
 
 """Replay Buffers Module."""
 
+from tf_agents.replay_buffers import episodic_replay_buffer
 from tf_agents.replay_buffers import py_hashed_replay_buffer
 from tf_agents.replay_buffers import py_uniform_replay_buffer
 from tf_agents.replay_buffers import replay_buffer
