Merge pull request #897 from cmarlin:init_replay_buffers

PiperOrigin-RevId: 591030872 Change-Id: Ibad67a777e4fe9e41ef61e239b5cca3e8bf0b5fa
tensorflow · Dec 14, 2023 · c8a7175 · c8a7175
2 parents afbe0b6 + c75494c
commit c8a7175
Showing 1 changed file with 1 addition and 0 deletions.
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